Spaces:
Sleeping
Sleeping
Benjamin Gonzalez
commited on
Commit
•
bc609bc
1
Parent(s):
fb38431
add changes
Browse files- packages.txt +0 -1
- pre-requirements.txt +1 -0
- requirements.txt +0 -1
packages.txt
DELETED
@@ -1 +0,0 @@
|
|
1 |
-
nvidia-cuda-toolkit
|
|
|
|
pre-requirements.txt
CHANGED
@@ -1,2 +1,3 @@
|
|
1 |
--extra-index-url https://download.pytorch.org/whl/cu113
|
2 |
torch==2.1.0
|
|
|
|
1 |
--extra-index-url https://download.pytorch.org/whl/cu113
|
2 |
torch==2.1.0
|
3 |
+
wheel
|
requirements.txt
CHANGED
@@ -1,4 +1,3 @@
|
|
1 |
-
flash-attn==2.3.6
|
2 |
mlflow==2.6.0
|
3 |
cloudpickle==2.2.1
|
4 |
jsonpickle==3.0.1
|
|
|
|
|
1 |
mlflow==2.6.0
|
2 |
cloudpickle==2.2.1
|
3 |
jsonpickle==3.0.1
|