Remove torchvision default dependency

Signed-off-by: Keval Morabia <[email protected]>

Author: kevalmorabia97

examples/cnn_qat/requirements.txt

@@ -0,0 +1 @@
+torchvision

examples/llm_eval/requirements.txt

@@ -4,3 +4,4 @@ openai>=0.28.1
 peft>=0.5.0
 rwkv>=0.7.3
 tiktoken
+torchvision

examples/onnx_ptq/requirements.txt

@@ -2,3 +2,4 @@ datasets>=2.14.4
 optimum
 sentencepiece
 timm
+torchvision

examples/pruning/cifar_resnet.ipynb

@@ -22,6 +22,15 @@
         "Let's first install `Model Optimizer` following the [installation steps](https://nvidia.github.io/TensorRT-Model-Optimizer/getting_started/2_installation.html)."
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "! pip install nvidia-modelopt torchvision"
+      ]
+    },
     {
       "cell_type": "code",
       "execution_count": 2,

setup.py

@@ -38,7 +38,6 @@
     "safetensors",
     "torch>=2.6",
     "torchprofile>=0.0.4",
-    "torchvision",
 ]
 
 optional_deps = {
@@ -79,6 +78,7 @@
         "pytest-cov",
         "pytest-timeout",
         "timm",
+        "torchvision",
         "tox>4.18",
         "tox-current-env>=0.0.12",
     ],