Merge branch 'Comfy-Org:main' into patch-3

Author: qiacheng

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/command/models/models.py

@@ -394,6 +394,11 @@ def remove(
         typer.echo("Deletion canceled.")
 
 
+def list_models(path: pathlib.Path) -> list:
+    """List all models in the specified directory."""
+    return [file for file in path.iterdir() if file.is_file()]
+
+
 @app.command()
 @tracking.track_command("model")
 def list(
@@ -416,8 +421,3 @@ def list(
     data = [(model.name, f"{model.stat().st_size // 1024} KB") for model in models]
     column_names = ["Model Name", "Size"]
     ui.display_table(data, column_names)
-
-
-def list_models(path: pathlib.Path) -> list:
-    """List all models in the specified directory."""
-    return [file for file in path.iterdir() if file.is_file()]

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"