xinjie.wang commited on
Commit
403bc07
·
1 Parent(s): 8571ded
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,9 +1,9 @@
1
  --extra-index-url https://download.pytorch.org/whl/cu121
2
 
3
 
4
- torch==2.4.0+cu121
5
- torchvision==0.19.0+cu121
6
- xformers==0.0.27.post2+cu121
7
  pytorch-lightning==2.4.0
8
  spconv-cu120==2.3.6
9
  dataclasses_json
 
1
  --extra-index-url https://download.pytorch.org/whl/cu121
2
 
3
 
4
+ torch==2.4.0
5
+ torchvision==0.19.0
6
+ xformers==0.0.27.post2
7
  pytorch-lightning==2.4.0
8
  spconv-cu120==2.3.6
9
  dataclasses_json