Seokju Cho commited on
Commit
c7a1328
1 Parent(s): a1a69e9

fix requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -3
requirements.txt CHANGED
@@ -2,6 +2,7 @@ einshape==1.0
2
  gradio==4.40.0
3
  mediapy==1.2.2
4
  opencv-python==4.10.0.84
5
- torch==2.4.0
6
- torchaudio==2.4.0
7
- torchvision==0.19.0
 
 
2
  gradio==4.40.0
3
  mediapy==1.2.2
4
  opencv-python==4.10.0.84
5
+ --extra-index-url https://download.pytorch.org/whl/cu121
6
+ torch==2.2.0
7
+ torchvision==0.17.0
8
+ torchaudio==2.2.0