Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
d90b45d
1
Parent(s):
2bc83bd
update dependency
Browse files- requirements.txt +4 -4
requirements.txt
CHANGED
@@ -1,14 +1,14 @@
|
|
1 |
omegaconf
|
2 |
-
torch
|
3 |
-
|
4 |
transformers==4.44.0
|
5 |
sentencepiece==0.1.99
|
6 |
accelerate==0.33.0
|
7 |
einops==0.6.1
|
8 |
einops-exts==0.0.4
|
9 |
timm==0.6.13
|
10 |
-
flash_attn
|
11 |
-
|
12 |
scipy
|
13 |
gradio
|
14 |
|
|
|
1 |
omegaconf
|
2 |
+
torch
|
3 |
+
torchvision
|
4 |
transformers==4.44.0
|
5 |
sentencepiece==0.1.99
|
6 |
accelerate==0.33.0
|
7 |
einops==0.6.1
|
8 |
einops-exts==0.0.4
|
9 |
timm==0.6.13
|
10 |
+
# flash_attn
|
11 |
+
https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.3/flash_attn-2.6.3+cu123torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|
12 |
scipy
|
13 |
gradio
|
14 |
|