Add ComfyUI Desktop worker support

2026-04-03 21:54:51 +00:00
parent dd55ff740e
commit e7ab67733b
5 changed files with 232 additions and 11 deletions
@@ -33,6 +33,7 @@ class ProcessLifecycle:
        env["CUDA_VISIBLE_DEVICES"] = str(worker_config.get("cuda_device", 0))
        env["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
        env["COMFYUI_MASTER_PID"] = str(os.getpid())
+        env["COMFYUI_IS_WORKER"] = "1"

        cmd = self._manager.build_launch_command(worker_config, comfy_root)
        cwd = comfy_root