Support CUDA 12.9 (#326)

Author: yt-koike

comfy_cli/command/install.py

@@ -72,7 +72,12 @@ def pip_install_comfyui_dependencies(
                 "torchaudio",
             ]
 
-            if plat == constants.OS.WINDOWS and cuda_version == constants.CUDAVersion.v12_6:
+            if plat == constants.OS.WINDOWS and cuda_version == constants.CUDAVersion.v12_9:
+                base_command += [
+                    "--extra-index-url",
+                    "https://download.pytorch.org/whl/cu129",
+                ]
+            elif plat == constants.OS.WINDOWS and cuda_version == constants.CUDAVersion.v12_6:
                 base_command += [
                     "--extra-index-url",
                     "https://download.pytorch.org/whl/cu126",

comfy_cli/constants.py

@@ -62,6 +62,7 @@ class PROC(str, Enum):
 
 
 class CUDAVersion(str, Enum):
+    v12_9 = "12.9"
     v12_6 = "12.6"
     v12_4 = "12.4"
     v12_1 = "12.1"