Launch comfy server with extra args specified in setting json

package.json

@@ -9,7 +9,7 @@
   "main": ".vite/build/main.js",
   "packageManager": "[email protected]",
   "config": {
-    "frontendVersion": "1.4.5",
+    "frontendVersion": "1.4.9",
     "comfyVersion": "0.3.2",
     "managerCommit": "1ab304bfbcfd7f550fc5adbb75a4e3b24861b947",
     "uvVersion": "0.5.2"

src/config/comfySettings.ts

@@ -9,11 +9,13 @@ export const DEFAULT_SETTINGS: ComfySettingsData = {
   'Comfy.UseNewMenu': 'Top',
   'Comfy.Workflow.WorkflowTabsPosition': 'Topbar',
   'Comfy.Workflow.ShowMissingModelsWarning': true,
+  'Comfy.Server.LaunchArgs': {},
 } as const;
 
 export interface ComfySettingsData {
   'Comfy-Desktop.AutoUpdate': boolean;
   'Comfy-Desktop.SendStatistics': boolean;
+  'Comfy.Server.LaunchArgs': Record<string, string | boolean>;
   [key: string]: unknown;
 }
 

src/constants.ts

@@ -104,7 +104,7 @@ export const DEFAULT_SERVER_ARGS = {
   /** The port to use for the ComfyUI server. */
   port: 8000,
   // Extra arguments to pass to the ComfyUI server.
-  extraServerArgs: {} as Record<string, string>,
+  extraServerArgs: {} as Record<string, string | boolean>,
 };
 
 export type ServerArgs = typeof DEFAULT_SERVER_ARGS;

src/main-process/comfyServer.ts

@@ -60,12 +60,12 @@ export class ComfyServer {
    */
   get coreLaunchArgs() {
     return {
-      '--user-directory': this.userDirectoryPath,
-      '--input-directory': this.inputDirectoryPath,
-      '--output-directory': this.outputDirectoryPath,
-      '--front-end-root': this.webRootPath,
-      '--extra-model-paths-config': ComfyServerConfig.configPath,
-      '--port': this.serverArgs.port.toString(),
+      'user-directory': this.userDirectoryPath,
+      'input-directory': this.inputDirectoryPath,
+      'output-directory': this.outputDirectoryPath,
+      'front-end-root': this.webRootPath,
+      'extra-model-paths-config': ComfyServerConfig.configPath,
+      port: this.serverArgs.port.toString(),
     };
   }
 
@@ -75,8 +75,10 @@ export class ComfyServer {
         ...this.serverArgs.extraServerArgs,
         ...this.coreLaunchArgs,
       })
+        .map(([key, value]) => [`--${key}`, value])
         .flat()
-        .filter((value: string) => !!value)
+        // Boolean true values are ignored. e.g. { '--cpu': true } => '--cpu'
+        .filter((value: string | boolean) => typeof value === 'string' && value !== '')
     );
   }
 

src/main.ts

@@ -85,7 +85,12 @@ if (!gotTheLock) {
       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 extraServerArgs: Record<string, string> = process.env.COMFYUI_CPU_ONLY === 'true' ? { '--cpu': '' } : {};
+      const cpuOnly: Record<string, string | boolean> =
+        process.env.COMFYUI_CPU_ONLY === 'true' ? { '--cpu': true } : {};
+      const extraServerArgs: Record<string, string | boolean> = {
+        ...comfyDesktopApp.comfySettings.get('Comfy.Server.LaunchArgs'),
+        ...cpuOnly,
+      };
 
       if (!useExternalServer) {
         await comfyDesktopApp.startComfyServer({ host, port, extraServerArgs });