EvanTHU commited on
Commit
4659009
·
verified ·
1 Parent(s): cbf9a02

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu117
2
  bitsandbytes==0.41.3.post2
3
  dataloader==2.0
4
  decord==0.6.0
@@ -35,10 +35,10 @@ Shapely==2.0.4
35
  shortuuid==1.0.13
36
  smplx==0.1.26
37
  tokenizers==0.13.3
38
- torch==2.0.0+cu117 --extra-index-url https://download.pytorch.org/whl/cu117
39
  flash_attn==2.4.2
40
  torch_xla==2.3.0
41
- torchvision==0.15.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117
42
  tqdm==4.66.1
43
  transformers==4.28.1
44
  trimesh==3.22.1
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu118
2
  bitsandbytes==0.41.3.post2
3
  dataloader==2.0
4
  decord==0.6.0
 
35
  shortuuid==1.0.13
36
  smplx==0.1.26
37
  tokenizers==0.13.3
38
+ torch==2.0.0+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
39
  flash_attn==2.4.2
40
  torch_xla==2.3.0
41
+ torchvision==0.15.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
42
  tqdm==4.66.1
43
  transformers==4.28.1
44
  trimesh==3.22.1