Update requirements.txt

2023-04-06 13:50:25 +02:00
parent 7b18b39dd8
commit 30bf938d03
1 changed files with 1 additions and 0 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,6 +4,7 @@ bitsandbytes
 datasets
 sentencepiece
 safetensors
+flash-attn
 git+https://github.com/huggingface/transformers.git
 git+https://github.com/sterlind/GPTQ-for-LLaMa.git@lora_4bit
 git+https://github.com/sterlind/peft.git