Hugging Face's logo

Spaces:
Iker
/
ClickbaitFighter
Running on Zero

App Files Community
Iker commited on
Commit
11b45da
·
verified ·
1 Parent(s): d3f082f

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -1
requirements.txt CHANGED
@@ -8,4 +8,6 @@ transformers==4.40.0
8
  torch
9
  sentencepiece
10
  accelerate
11
- flash-attn
 
 
 
8
  torch
9
  sentencepiece
10
  accelerate
11
+ flash-attn
12
+ torch==2.2
13
+ https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.2/flash_attn-2.6.2+cu118torch2.2cxx11abiFALSE-cp311-cp311-linux_x86_64.whl