diff --git a/comfy/execution_context.py b/comfy/execution_context.py index 15a78f97d..29bb89535 100644 --- a/comfy/execution_context.py +++ b/comfy/execution_context.py @@ -18,7 +18,10 @@ _empty_execution_context = ExecutionContext(ServerStub()) def current_execution_context() -> ExecutionContext: - return _current_context.get() + try: + return _current_context.get() + except LookupError: + return _empty_execution_context @contextmanager