diff --git a/comfy_cli/command/install.py b/comfy_cli/command/install.py index 7c21e26..ad8380a 100755 --- a/comfy_cli/command/install.py +++ b/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", diff --git a/comfy_cli/constants.py b/comfy_cli/constants.py index 75af1a8..88ef415 100644 --- a/comfy_cli/constants.py +++ b/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"