Merge pull request #22 from NVIDIA-NeMo/pagaray_add_for_automodel_dep

feat: add for automodel dep

Author: pablo-garay

pyproject.toml

@@ -66,7 +66,7 @@ automodel = [
     "diffusers",
     "ftfy",
     "imageio-ffmpeg",
-    "opencv-python-headless==4.10.0.84"
+    "opencv-python-headless==4.10.0.84",
 ]
 docs = [
     "myst-parser>=4.0.1",
@@ -97,6 +97,22 @@ packages = ["nemo_vfm"]
 version = {attr = "nemo_vfm.__version__"}
 readme = {file = "README.md", content-type = "text/markdown"}
 
+[[tool.uv.index]]
+name = "pypi"
+url = "https://pypi.org/simple"
+explicit = true
+
+[[tool.uv.index]]
+name = "pytorch"
+url = "https://download.pytorch.org/whl/cu128"
+explicit = true
+
+[tool.uv.sources]
+torch = [
+    { index = "pytorch", marker = "sys_platform != 'darwin'" },
+    { index = "pypi", marker = "sys_platform == 'darwin'" },
+]
+
 [project.urls]
 Download = "https://github.com/NVIDIA-NeMo/NeMo-VFM/releases"
 Homepage = "https://github.com/NVIDIA-NeMo/NeMo-VFM"