Add CPU option for install.

Comfy-Org · May 8, 2024 · b1d001c · b1d001c
1 parent 9c7f68d
commit b1d001c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/.github/workflows/build-and-test.yml b/.github/workflows/build-and-test.yml
@@ -32,4 +32,4 @@ jobs:
       - name: Run Tests
         run: |
           comfy --yes env
-          comfy install
+          comfy install --cpu
diff --git a/comfy_cli/cmdline.py b/comfy_cli/cmdline.py
@@ -155,6 +155,14 @@ def install(
             callback=gpu_exclusivity_callback,
         ),
     ] = None,
+    cpu: Annotated[
+        bool,
+        typer.Option(
+            show_default=False,
+            help="Install for CPU",
+            callback=gpu_exclusivity_callback,
+        ),
+    ] = None,
     commit: Annotated[
         Optional[str], typer.Option(help="Specify commit hash for ComfyUI")
     ] = None,
@@ -206,6 +214,8 @@ def install(
         gpu = GPU_OPTION.AMD
     elif m_series:
         gpu = GPU_OPTION.M_SERIES
+    elif cpu:
+        gpu = None
     else:
         if platform == constants.OS.MACOS:
             gpu = ui.prompt_select_enum(