jamesliu1217 commited on
Commit
17f9e9d
·
verified ·
1 Parent(s): 411cd1a

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -4
requirements.txt CHANGED
@@ -1,7 +1,6 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu124
2
- torch
3
- torchvision
4
- torchaudio
5
  diffusers==0.32.2
6
  easydict==1.13
7
  einops==0.8.1
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu118
2
+ torch==2.3.1
3
+ torchvision==0.18.1
 
4
  diffusers==0.32.2
5
  easydict==1.13
6
  einops==0.8.1