Update requirements.txt
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
@@ -10,6 +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
|
|
|
10 |
git+https://github.com/huggingface/transformers@3a1ead0aabed473eafe527915eea8c197d424356
|
11 |
accelerate
|
12 |
av
|
13 |
+
torchvision
|
14 |
|
15 |
# Optional dependency
|
16 |
# Uncomment the following line if you need flash-attn
|