mfarre HF staff commited on
Commit
4b68e47
·
1 Parent(s): 141829b

using transformers to handle the model

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -3,5 +3,6 @@ opencv-python
3
  num2words
4
  ffmpeg-python
5
  transformers @ git+https://github.com/huggingface/transformers.git@refs/pull/36126/head
 
6
  accelerate>=0.26.0
7
  decord==0.6.0
 
3
  num2words
4
  ffmpeg-python
5
  transformers @ git+https://github.com/huggingface/transformers.git@refs/pull/36126/head
6
+ flash-attn
7
  accelerate>=0.26.0
8
  decord==0.6.0