Update requirements.txt
Browse files- requirements.txt +3 -1
requirements.txt
CHANGED
|
@@ -10,7 +10,9 @@ modelscope_studio==1.2.2
|
|
| 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
|
|
|
|
| 10 |
git+https://github.com/huggingface/transformers@3a1ead0aabed473eafe527915eea8c197d424356
|
| 11 |
accelerate
|
| 12 |
av
|
| 13 |
+
torch # Install torch first
|
| 14 |
torchvision
|
| 15 |
+
|
| 16 |
# Optional dependency
|
| 17 |
# Uncomment the following line if you need flash-attn
|
| 18 |
+
flash-attn==2.7.4.post1
|