Spaces:
Runtime error
Runtime error
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
|