Add torchvision dependency to resolve nms issue
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
@@ -1,7 +1,8 @@
|
|
1 |
torch>=2.0.0
|
|
|
2 |
transformers>=4.25.0
|
3 |
unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git
|
4 |
-
xformers==0.0.27
|
5 |
trl
|
6 |
peft
|
7 |
accelerate
|
|
|
1 |
torch>=2.0.0
|
2 |
+
torchvision
|
3 |
transformers>=4.25.0
|
4 |
unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git
|
5 |
+
xformers==0.0.27
|
6 |
trl
|
7 |
peft
|
8 |
accelerate
|