worker launch bug fix

2026-03-07 05:45:11 +00:00
parent ce3b2c63f3
commit ab4e1536a5
1 changed files with 1 additions and 0 deletions
@@ -39,6 +39,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