Spaces:
Sleeping
Sleeping
Benjamin Gonzalez
commited on
Commit
•
f5cca13
1
Parent(s):
0151e30
update
Browse files- pre-requirements.txt +2 -0
- 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
|