Spaces:
Running
Running
Update requirements.txt
Browse files- requirements.txt +6 -7
requirements.txt
CHANGED
|
@@ -1,10 +1,9 @@
|
|
| 1 |
--index-url https://download.pytorch.org/whl/cu121
|
| 2 |
-
torch
|
| 3 |
-
torchvision
|
| 4 |
-
torchaudio
|
| 5 |
-
huggingface_hub
|
| 6 |
-
transformers
|
| 7 |
-
peft
|
| 8 |
-
flash-attn
|
| 9 |
wheel
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu122torch2.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|
|
|
|
| 1 |
--index-url https://download.pytorch.org/whl/cu121
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
wheel
|
| 3 |
+
torch==2.3.1
|
| 4 |
+
torchvision==0.18.1
|
| 5 |
+
torchaudio==2.3.1
|
| 6 |
+
huggingface_hub==0.23.3
|
| 7 |
+
transformers==4.41.2
|
| 8 |
+
peft==0.11.1
|
| 9 |
https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu122torch2.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|