1MR commited on
Commit
b9da150
·
verified ·
1 Parent(s): dc47049

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -5,3 +5,4 @@ sentencepiece==0.1.96
5
  torch==1.11.0
6
  transformers==4.21.1
7
  numpy==1.24.4 # Ensure compatibility
 
 
5
  torch==1.11.0
6
  transformers==4.21.1
7
  numpy==1.24.4 # Ensure compatibility
8
+ flash_attn