jamesliu1217 commited on
Commit
e403b1b
·
verified ·
1 Parent(s): f55a7b1

Update requirements.txt

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