Merge pull request #20 from NVIDIA-NeMo/teamwork-groupforce/dependency-add

feat: update pyproject.toml

Author: akoumpa

pyproject.toml

@@ -61,6 +61,23 @@ requires = ["setuptools>=61"]
 build-backend = "setuptools.build_meta"
 
 [dependency-groups]
+automodel = [
+    "bitsandbytes==0.45.5; (platform_machine == 'x86_64' and platform_system != 'Darwin')",
+    "datasets>=4.0.0",
+    "diffusers",
+    "ftfy",
+    "imageio-ffmpeg",
+    "liger-kernel>=0.5.9; (platform_machine == 'x86_64' and platform_system != 'Darwin')",
+    "megatron-fsdp",
+    "opencv-python-headless==4.10.0.84",
+    "pybind11",
+    "pyyaml",
+    "torch<=2.8.0",
+    "torchdata",
+    "transformers<=4.57.1",
+    "wandb",
+    "torchao",
+]
 docs = [
     "myst-parser>=4.0.1",
     "nvidia-sphinx-theme>=0.0.8",