Zeph27 commited on
Commit
9bfb04d
Β·
1 Parent(s): 28bbaa0

flash attn

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -5,4 +5,5 @@ transformers==4.40.0
5
  sentencepiece==0.1.99
6
  accelerate==0.30.1
7
  bitsandbytes==0.43.1
8
- decord
 
 
5
  sentencepiece==0.1.99
6
  accelerate==0.30.1
7
  bitsandbytes==0.43.1
8
+ decord
9
+ flash_attn