jdesiree commited on
Commit
86709fb
·
verified ·
1 Parent(s): 6a7529b

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -12,6 +12,7 @@ torch>=2.0.0
12
  transformers>=4.41.2
13
  accelerate>=0.30.0
14
  bitsandbytes>=0.43.0
 
15
 
16
  # Utilities
17
  python-dotenv>=1.0.0
 
12
  transformers>=4.41.2
13
  accelerate>=0.30.0
14
  bitsandbytes>=0.43.0
15
+ flash-attn
16
 
17
  # Utilities
18
  python-dotenv>=1.0.0