Yoxas commited on
Commit
11431ac
·
verified ·
1 Parent(s): 65df291

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -2,4 +2,4 @@ huggingface_hub==0.22.2
2
  transformers
3
  torch
4
  sentencepiece
5
- flash-attn
 
2
  transformers
3
  torch
4
  sentencepiece
5
+ flash-attn[CUDA12]