Benjamin Gonzalez commited on
Commit
8377710
1 Parent(s): 5315eed

add repo to requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -1,3 +1,4 @@
 
1
  flash-attn==2.3.6
2
  mlflow==2.6.0
3
  cloudpickle==2.2.1
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu113
2
  flash-attn==2.3.6
3
  mlflow==2.6.0
4
  cloudpickle==2.2.1