fix: --fast-deps now respects --skip-torch-or-directml (#246)

Author: bigcat88

comfy_cli/command/install.py

@@ -287,6 +287,7 @@ def execute(
             gpu=gpu,
             cuda_version=cuda_version.value,
             rocm_version=rocm_version.value,
+            skip_torch=skip_torch_or_directml,
         )
         depComp.compile_deps()
         depComp.install_deps()

comfy_cli/uv.py

@@ -367,6 +367,7 @@ def __init__(
         extraSpecs: list[str] | None = None,
         cuda_version: str | None = None,
         rocm_version: str | None = None,
+        skip_torch: bool = False,
     ):
         """Compiler/installer of Python dependencies based on uv
 
@@ -379,16 +380,21 @@ def __init__(
             reqFilesCore (Optional[list[PathLike]]): list of core requirement files (requirements.txt, pyproject.toml, etc) to be included in the compilation. Any requirements determined from these files will override all other requirements
             reqFilesExt (Optional[list[PathLike]]): list of requirement files (requirements.txt, pyproject.toml, etc) to be included in the compilation
             extraSpecs (Optional[list[str]]): list of extra Python requirement specifiers to be included in the compilation
+            skip_torch (bool): if True, skip torch/torchvision/torchaudio installation and GPU index URLs
         """
         self.cwd = Path(cwd).expanduser().resolve()
         self.outDir: Path = Path(outDir).expanduser().resolve()
         # use .absolute since .resolve breaks the softlink-is-interpreter assumption of venvs
         self.executable = Path(executable).expanduser().absolute()
         self.gpu = DependencyCompiler.Resolve_Gpu(gpu)
+        self.skip_torch = skip_torch
         self.reqFiles = [Path(reqFile) for reqFile in reqFilesExt] if reqFilesExt is not None else None
         self.extraSpecs = [] if extraSpecs is None else extraSpecs
 
-        if self.gpu == GPU_OPTION.NVIDIA:
+        if self.skip_torch:
+            self.gpuUrl = None
+            self.torchBackend = None
+        elif self.gpu == GPU_OPTION.NVIDIA:
             tag = f"cu{cuda_version.replace('.', '')}" if cuda_version else DependencyCompiler.nvidiaTorchBackend
             self.gpuUrl = f"https://download.pytorch.org/whl/{tag}"
             self.torchBackend = tag

tests/uv/test_uv.py

@@ -177,3 +177,35 @@ def test_amd_custom_rocm_version():
     )
     assert depComp.torchBackend == "rocm7.1"
     assert depComp.gpuUrl == "https://download.pytorch.org/whl/rocm7.1"
+
+
+@pytest.mark.parametrize("gpu", [GPU_OPTION.NVIDIA, GPU_OPTION.AMD, GPU_OPTION.CPU])
+def test_skip_torch_disables_gpu_url_and_backend(gpu):
+    depComp = DependencyCompiler(cwd=temp, gpu=gpu, outDir=temp, reqFilesCore=[], reqFilesExt=[], skip_torch=True)
+    assert depComp.torchBackend is None
+    assert depComp.gpuUrl is None
+
+
+def test_skip_torch_override_has_no_torch():
+    depComp = DependencyCompiler(
+        cwd=temp,
+        gpu=GPU_OPTION.NVIDIA,
+        outDir=temp,
+        reqFilesCore=[mockReqsDir / "core_reqs.txt"],
+        reqFilesExt=[],
+        skip_torch=True,
+    )
+    depComp.make_override()
+    content = depComp.override.read_text()
+    assert "torch" not in content
+
+
+def test_skip_torch_install_deps_no_extra_index_url():
+    depComp = DependencyCompiler(
+        cwd=temp, gpu=GPU_OPTION.NVIDIA, outDir=temp, reqFilesCore=[], reqFilesExt=[], skip_torch=True
+    )
+    depComp.out.write_text("requests==2.31.0\n")
+    with patch("comfy_cli.uv._check_call") as mock_check_call:
+        depComp.install_deps()
+    cmd = mock_check_call.call_args[0][0]
+    assert "--extra-index-url" not in cmd