samdeniyi commited on
Commit
92dc33a
1 Parent(s): f89ab5e

Add torchvision dependency to resolve nms issue

Browse files
Files changed (1) hide show
  1. 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 # or the appropriate version for your Torch version
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