feat: support --commit in comfy install

#11

Author: ltdrdata

comfy_cli/cmdline.py

@@ -79,14 +79,16 @@ def install(
   ] = False,
   skip_manager: Annotated[
     bool,
-    typer.Option(
-      help="Skip installing the manager component")
+    typer.Option(help="Skip installing the manager component")
   ] = False,
   amd: Annotated[
     bool,
-    typer.Option(
-      help="Install for AMD gpu")
+    typer.Option(help="Install for AMD gpu")
   ] = False,
+  commit: Annotated[
+    str,
+    typer.Option(help="Specify commit hash for ComfyUI")
+  ] = None
 ):
   checker = EnvChecker()
 
@@ -119,7 +121,7 @@ def install(
   if amd:
     torch_mode = 'amd'
 
-  install_inner.execute(url, manager_url, workspace_path, restore, skip_manager, torch_mode)
+  install_inner.execute(url, manager_url, workspace_path, restore, skip_manager, torch_mode, commit=commit)
   workspace_manager.set_recent_workspace(workspace_path)
 
 

comfy_cli/command/install.py

@@ -24,8 +24,8 @@ def install_manager_dependencies(repo_dir):
   subprocess.run([sys.executable, '-m', "pip", "install", "-r", "requirements.txt"])
 
 
-def execute(url: str, manager_url: str, comfy_workspace: str, restore: bool, skip_manager: bool, torch_mode=None, *args,
-            **kwargs):
+def execute(url: str, manager_url: str, comfy_workspace: str, restore: bool, skip_manager: bool, torch_mode=None, commit=None,
+            *args, **kwargs):
   print(f"Installing from {url}")
 
   # install ComfyUI
@@ -34,7 +34,11 @@ def execute(url: str, manager_url: str, comfy_workspace: str, restore: bool, ski
   repo_dir = os.path.abspath(repo_dir)
 
   if os.path.exists(os.path.join(repo_dir, '.git')):
-    if restore:
+    if restore or commit is not None:
+      if commit is not None:
+        os.chdir(repo_dir)
+        subprocess.run(["git", "checkout", commit])
+
       install_comfyui_dependencies(repo_dir, torch_mode)
     else:
       print(
@@ -47,6 +51,11 @@ def execute(url: str, manager_url: str, comfy_workspace: str, restore: bool, ski
     repo_dir = os.path.abspath(repo_dir)
     subprocess.run(["git", "clone", url, repo_dir])
 
+    # checkout specified commit
+    if commit is not None:
+      os.chdir(repo_dir)
+      subprocess.run(["git", "checkout", commit])
+
     install_comfyui_dependencies(repo_dir, torch_mode)
 
   print("")