Spaces:
Paused
Paused
Update requirements.txt
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
@@ -4,6 +4,7 @@ torch
|
|
4 |
einops
|
5 |
accelerate
|
6 |
tiktoken
|
|
|
7 |
transformers_stream_generator==0.0.4
|
8 |
peft
|
9 |
deepspeed
|
|
|
4 |
einops
|
5 |
accelerate
|
6 |
tiktoken
|
7 |
+
flash-attention
|
8 |
transformers_stream_generator==0.0.4
|
9 |
peft
|
10 |
deepspeed
|