From 73a11c0dbb037b9c410b7f6acd289cbf8bad0d98 Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Wed, 12 Jun 2024 14:34:24 -0700
Subject: [PATCH] Fix execution context interaction

---
 comfy/execution_context.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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