Fix .env CPU flag ignored (#449)

Comfy-Org · Dec 7, 2024 · 529bb1f · 529bb1f
1 parent c0ead0f
commit 529bb1f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/main-process/comfyServer.ts b/src/main-process/comfyServer.ts
@@ -83,7 +83,7 @@ export class ComfyServer {
       Object.entries(args)
         .map(([key, value]) => [`--${key}`, value])
         .flat()
-        // Empty string values are ignored. e.g. { '--cpu': '' } => '--cpu'
+        // Empty string values are ignored. e.g. { cpu: '' } => '--cpu'
         .filter((value: string) => value !== '')
     );
   }

diff --git a/src/main.ts b/src/main.ts
@@ -76,7 +76,7 @@ async function startApp() {
       const host = process.env.COMFY_HOST || DEFAULT_SERVER_ARGS.host;
       const targetPort = process.env.COMFY_PORT ? parseInt(process.env.COMFY_PORT) : DEFAULT_SERVER_ARGS.port;
       const port = useExternalServer ? targetPort : await findAvailablePort(host, targetPort, targetPort + 1000);
-      const cpuOnly: Record<string, string> = process.env.COMFYUI_CPU_ONLY === 'true' ? { '--cpu': '' } : {};
+      const cpuOnly: Record<string, string> = process.env.COMFYUI_CPU_ONLY === 'true' ? { cpu: '' } : {};
       const extraServerArgs: Record<string, string> = {
         ...comfyDesktopApp.comfySettings.get('Comfy.Server.LaunchArgs'),
         ...cpuOnly,