Spaces:
Running
on
Zero
Running
on
Zero
Update requirements.txt
Browse files- requirements.txt +3 -2
requirements.txt
CHANGED
@@ -42,8 +42,9 @@ onnxruntime
|
|
42 |
modelscope
|
43 |
word2number
|
44 |
|
45 |
-
torch
|
46 |
-
|
|
|
47 |
https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|
48 |
|
49 |
|
|
|
42 |
modelscope
|
43 |
word2number
|
44 |
|
45 |
+
torch==2.6.0
|
46 |
+
torchvision==0.21.0
|
47 |
+
torchaudio==2.6.0
|
48 |
https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|
49 |
|
50 |
|