ashawkey commited on
Commit
902e84b
1 Parent(s): bb85cbc
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
- torch==2.2.0 --index-url https://download.pytorch.org/whl/cu118
2
- xformers --index-url https://download.pytorch.org/whl/cu118
3
 
4
  numpy
5
  tyro
 
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