Spaces:
Sleeping
Sleeping
Plat
commited on
Commit
·
3558ff7
1
Parent(s):
00d5978
chore: add flash-attn
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
@@ -2,4 +2,5 @@ torch==2.2.0
|
|
2 |
accelerate==0.30.1
|
3 |
transformers==4.41.2
|
4 |
spaces==0.28.3
|
5 |
-
bitsandbytes==0.43.1
|
|
|
|
2 |
accelerate==0.30.1
|
3 |
transformers==4.41.2
|
4 |
spaces==0.28.3
|
5 |
+
bitsandbytes==0.43.1
|
6 |
+
flash-attn==2.5.9 --no-build-isolation
|