Benjamin Gonzalez commited on
Commit
f5cca13
1 Parent(s): 0151e30
Files changed (2) hide show
  1. pre-requirements.txt +2 -0
  2. requirements.txt +0 -2
pre-requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu113
2
+ torch==2.1.0
requirements.txt CHANGED
@@ -1,5 +1,3 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu113
2
- torch==2.1.0
3
  flash-attn==2.3.6
4
  mlflow==2.6.0
5
  cloudpickle==2.2.1
 
 
 
1
  flash-attn==2.3.6
2
  mlflow==2.6.0
3
  cloudpickle==2.2.1