dwa

Comfy-Org · Jul 3, 2024 · d71b31d · d71b31d
1 parent 65bea8e
commit d71b31d
Showing 1 changed file with 12 additions and 13 deletions.
diff --git a/comfy_cli/command/install.py b/comfy_cli/command/install.py
@@ -51,34 +51,33 @@ def install_comfyui_dependencies(
 
         # install torch for NVIDIA
         if gpu == GPU_OPTION.NVIDIA:
-            pip_url = ""
+            base_command = [
+                sys.executable,
+                "-m",
+                "pip",
+                "install",
+                "torch",
+                "torchvision",
+                "torchaudio",
+            ]
             if (
                 plat == constants.OS.WINDOWS
                 and cuda_version == constants.CUDAVersion.v12_1
             ):
-                pip_url = [
+                base_command += [
                     "--extra-index-url",
                     "https://download.pytorch.org/whl/cu121",
                 ]
             elif (
                 plat == constants.OS.WINDOWS
                 and cuda_version == constants.CUDAVersion.v11_8
             ):
-                pip_url = [
+                base_command += [
                     "--extra-index-url",
                     "https://download.pytorch.org/whl/cu118",
                 ]
             result = subprocess.run(
-                [
-                    sys.executable,
-                    "-m",
-                    "pip",
-                    "install",
-                    "torch",
-                    "torchvision",
-                    "torchaudio",
-                ]
-                + pip_url,
+                base_command,
                 check=False,
             )
         # Beta support for intel arch based on this PR: https://github.com/comfyanonymous/ComfyUI/pull/3439