Skip to content

GitLab

"...resnet50_tensorflow.git" did not exist on "a4ae812e57a14108870f295a9e87c5a5a9f2a73f"
Commit 314d28c2 authored by comfyanonymous's avatar comfyanonymous
Browse files

Pass extra_pnginfo as None when not in input data.

parent 55f37baa
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
execution.py
View file @ 314d28c2
...@@ -35,8 +35,7 @@ def get_input_data(inputs, class_def, unique_id, outputs={}, prompt={}, extra_da ...@@ -35,8 +35,7 @@ def get_input_data(inputs, class_def, unique_id, outputs={}, prompt={}, extra_da
if h[x] == "PROMPT": if h[x] == "PROMPT":
input_data_all[x] = [prompt] input_data_all[x] = [prompt]
if h[x] == "EXTRA_PNGINFO": if h[x] == "EXTRA_PNGINFO":
if "extra_pnginfo" in extra_data: input_data_all[x] = [extra_data.get('extra_pnginfo', None)]
input_data_all[x] = [extra_data['extra_pnginfo']]
if h[x] == "UNIQUE_ID": if h[x] == "UNIQUE_ID":
input_data_all[x] = [unique_id] input_data_all[x] = [unique_id]
return input_data_all return input_data_all
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment