build: Move torchvision to optional package

Signed-off-by: Brandon Groth <[email protected]>

Author: BrandonGroth

fms_mo/utils/import_utils.py

@@ -31,6 +31,7 @@
     "pygraphviz",
     "fms",
     "triton",
+    "torchvision",
 ]
 
 available_packages = {}

pyproject.toml

@@ -32,7 +32,6 @@ dependencies = [
 "ninja>=1.11.1.1,<2.0",
 "tensorboard",
 "notebook",
-"torchvision>=0.17",
 "evaluate",
 "huggingface_hub",
 "pandas",
@@ -46,6 +45,7 @@ dev = ["pre-commit>=3.0.4,<5.0"]
 fp8 = ["llmcompressor"]
 gptq = ["Cython", "gptqmodel>=1.7.3"]
 mx = ["microxcaling>=1.1"]
+vision = ["torchvision>=0.17"]
 visualize = ["matplotlib", "graphviz", "pygraphviz"]
 flash-attn = ["flash-attn>=2.5.3,<3.0"]
 opt = ["fms-model-optimizer[fp8, gptq]"]