using transformers to handle the model
Browse files- 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
|