Update requirements.txt
Browse files- requirements.txt +2 -4
requirements.txt
CHANGED
|
@@ -1,10 +1,8 @@
|
|
| 1 |
-
numpy==1.26.4
|
| 2 |
Pillow==10.1.0
|
|
|
|
| 3 |
torch==2.1.2
|
| 4 |
torchvision==0.16.2
|
| 5 |
transformers==4.40.0
|
| 6 |
-
deepspeed==0.12.6
|
| 7 |
sentencepiece==0.1.99
|
| 8 |
gradio
|
| 9 |
-
decord
|
| 10 |
-
flash_attn
|
|
|
|
|
|
|
| 1 |
Pillow==10.1.0
|
| 2 |
+
flash_attn
|
| 3 |
torch==2.1.2
|
| 4 |
torchvision==0.16.2
|
| 5 |
transformers==4.40.0
|
|
|
|
| 6 |
sentencepiece==0.1.99
|
| 7 |
gradio
|
| 8 |
+
decord
|
|
|