YaohuiW commited on
Commit
ebfc68f
·
verified ·
1 Parent(s): 48b8c3d

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -2
requirements.txt CHANGED
@@ -1,10 +1,12 @@
 
 
 
 
1
  einops==0.8.1
2
  imageio==2.37.0
3
  numpy==1.24.3
4
  Pillow==11.3.0
5
  PyYAML==6.0.2
6
- torch==2.0.1
7
- torchvision==0.15.2
8
  tqdm==4.67.1
9
  ninja==1.11.1.4
10
  av==14.2.0
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu118
2
+ torch==2.0.1
3
+ torchvision==0.15.2
4
+
5
  einops==0.8.1
6
  imageio==2.37.0
7
  numpy==1.24.3
8
  Pillow==11.3.0
9
  PyYAML==6.0.2
 
 
10
  tqdm==4.67.1
11
  ninja==1.11.1.4
12
  av==14.2.0