Spaces:
Running
on
Zero
Running
on
Zero
wenxiang guo
commited on
Update requirements.txt
Browse files- requirements.txt +6 -1
requirements.txt
CHANGED
@@ -1,3 +1,7 @@
|
|
|
|
|
|
|
|
|
|
1 |
torchlibrosa==0.1.0
|
2 |
pytorch-lightning==2.3.0
|
3 |
jupyter==1.0.0
|
@@ -17,4 +21,5 @@ torchdyn==1.0.6
|
|
17 |
ninja==1.11.1.1
|
18 |
protobuf==5.27.1
|
19 |
openai==1.34.0
|
20 |
-
|
|
|
|
1 |
+
--extra-index-url https://download.pytorch.org/whl/cu118
|
2 |
+
torch==2.1.0
|
3 |
+
torchvision==0.16.0
|
4 |
+
torchaudio==2.1.0
|
5 |
torchlibrosa==0.1.0
|
6 |
pytorch-lightning==2.3.0
|
7 |
jupyter==1.0.0
|
|
|
21 |
ninja==1.11.1.1
|
22 |
protobuf==5.27.1
|
23 |
openai==1.34.0
|
24 |
+
numpy==1.23.5
|
25 |
+
flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|