yeq6x commited on
Commit
d2c5c96
·
1 Parent(s): f611314

dependencies

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu121
2
- torch==2.2.0
 
3
  transformers==4.47.1
4
  diffusers==0.31.0
5
  accelerate==1.2.1
 
1
+ -f https://download.pytorch.org/whl/cu118
2
+ torch==2.0.0
3
+ torchvision==0.15.1
4
  transformers==4.47.1
5
  diffusers==0.31.0
6
  accelerate==1.2.1