WilliamGazeley commited on
Commit
98b9e59
·
1 Parent(s): 646806e

Add flash attention

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -22,3 +22,4 @@ yfinance==0.2.36
22
  transformers==4.40.2
23
  langchain==0.1.9
24
  accelerate==0.27.2
 
 
22
  transformers==4.40.2
23
  langchain==0.1.9
24
  accelerate==0.27.2
25
+ flash-attn==2.5.5