ashawkey commited on
Commit
96cb920
1 Parent(s): 902e84b
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
- torch==2.2.0 --index-url https://download.pytorch.org/whl/cu121
2
- xformers --index-url https://download.pytorch.org/whl/cu121
 
3
 
4
  numpy
5
  tyro
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu118
2
+ torch==2.2.0
3
+ xformers
4
 
5
  numpy
6
  tyro