From 314d28c251a69ca1cb42f888635e66ceeba5a03a Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Thu, 7 Mar 2024 15:07:47 -0500
Subject: [PATCH] Pass extra_pnginfo as None when not in input data.

---
 execution.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/execution.py b/execution.py
index 3e9d53b0..5f8ea835 100644
--- a/execution.py
+++ b/execution.py
@@ -35,8 +35,7 @@ def get_input_data(inputs, class_def, unique_id, outputs={}, prompt={}, extra_da
             if h[x] == "PROMPT":
                 input_data_all[x] = [prompt]
             if h[x] == "EXTRA_PNGINFO":
-                if "extra_pnginfo" in extra_data:
-                    input_data_all[x] = [extra_data['extra_pnginfo']]
+                input_data_all[x] = [extra_data.get('extra_pnginfo', None)]
             if h[x] == "UNIQUE_ID":
                 input_data_all[x] = [unique_id]
     return input_data_all