Spaces:
Running
Running
Update requirements.txt
Browse files- requirements.txt +3 -3
requirements.txt
CHANGED
@@ -1,9 +1,9 @@
|
|
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
|
|
|
1 |
--index-url https://download.pytorch.org/whl/cu121
|
|
|
2 |
torch==2.3.1
|
3 |
torchvision==0.18.1
|
4 |
torchaudio==2.3.1
|
5 |
+
wheel==0.43.0
|
6 |
+
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
|
7 |
huggingface_hub==0.23.3
|
8 |
transformers==4.41.2
|
9 |
+
peft==0.11.1
|
|