add torch torchvision accelerate versions

This commit is contained in:
rockerBOO
2024-11-29 15:57:31 -05:00
parent c7cadbc8c7
commit 2dd063a679

View File

@@ -40,6 +40,7 @@ jobs:
run: |
python -m pip install --upgrade pip
pip install -r requirements.txt
pip install torch==2.4.0 torchvision==0.19.0 accelerate==0.33.0
- name: Test with pytest
run: |
pip install pytest pytest-cov