Spaces:
Running
on
A100
Running
on
A100
using transformers to handle the model
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
|
@@ -2,6 +2,7 @@ Pillow
|
|
| 2 |
opencv-python
|
| 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
|
|
|
|
| 2 |
opencv-python
|
| 3 |
num2words
|
| 4 |
ffmpeg-python
|
| 5 |
+
torch
|
| 6 |
transformers @ git+https://github.com/huggingface/transformers.git@refs/pull/36126/head
|
| 7 |
flash-attn
|
| 8 |
accelerate>=0.26.0
|