Update requirements.txt
Browse files- requirements.txt +2 -2
requirements.txt
CHANGED
|
@@ -10,7 +10,7 @@ modelscope_studio==1.2.2
|
|
| 10 |
git+https://github.com/huggingface/transformers@3a1ead0aabed473eafe527915eea8c197d424356
|
| 11 |
accelerate
|
| 12 |
av
|
| 13 |
-
|
| 14 |
# Optional dependency
|
| 15 |
# Uncomment the following line if you need flash-attn
|
| 16 |
-
|
|
|
|
| 10 |
git+https://github.com/huggingface/transformers@3a1ead0aabed473eafe527915eea8c197d424356
|
| 11 |
accelerate
|
| 12 |
av
|
| 13 |
+
torchvision
|
| 14 |
# Optional dependency
|
| 15 |
# Uncomment the following line if you need flash-attn
|
| 16 |
+
flash-attn==2.7.4.post1
|