Spaces:
Sleeping
Sleeping
Commit
·
bb17d35
1
Parent(s):
72ab313
modify requirements
Browse files- pre-requirements.txt +3 -0
- requirements.txt +0 -3
pre-requirements.txt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
torch==2.1.2 --index-url https://download.pytorch.org/whl/cu121
|
2 |
+
torchvision==0.16.2 --index-url https://download.pytorch.org/whl/cu121
|
3 |
+
torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu121
|
requirements.txt
CHANGED
@@ -35,9 +35,6 @@ safetensors==0.4.3
|
|
35 |
scikit-learn==1.2.2
|
36 |
scipy==1.11.1
|
37 |
sentencepiece==0.1.99
|
38 |
-
torch==2.1.2 --index-url https://download.pytorch.org/whl/cu121
|
39 |
-
torchvision==0.16.2 --index-url https://download.pytorch.org/whl/cu121
|
40 |
-
torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu121
|
41 |
flash-attn==2.4.2
|
42 |
timm==0.9.11
|
43 |
toml==0.10.2
|
|
|
35 |
scikit-learn==1.2.2
|
36 |
scipy==1.11.1
|
37 |
sentencepiece==0.1.99
|
|
|
|
|
|
|
38 |
flash-attn==2.4.2
|
39 |
timm==0.9.11
|
40 |
toml==0.10.2
|