lukas-judith commited on
Commit
7933812
1 Parent(s): b034b83

Fix dependency issues

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -4
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
- accelerate==0.22.0
2
- torch==2.0.1
3
- torchvision==0.15.2
4
- transformers[sklearn,sentencepiece,audio,vision]==4.32.1
5
  diffusers==0.20.2
6
  --extra-index-url https://download.pytorch.org/whl/cu118
 
1
+ accelerate==0.31.0
2
+ torch==2.2.0
3
+ torchvision==0.17.0
4
+ transformers[sklearn,sentencepiece,audio,vision]==4.38.2
5
  diffusers==0.20.2
6
  --extra-index-url https://download.pytorch.org/whl/cu118