wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-15 16:02:32 +08:00
Code Issues Actions 20 Packages Projects Releases Wiki Activity

Added checks to ensure prompt is a JSON dict

Browse Source
This commit is contained in:
Blake C 2023-11-23 23:24:11 -08:00
parent 982338b9bb
commit 5a7e063122
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
execution.py
View File

@ -196,6 +196,10 @@ def recursive_execute(server, prompt, outputs, current_item, extra_data, execute
return (True, None, None)
def recursive_will_execute(prompt, outputs, current_item):
if type(prompt) is not dict:
prompt = json.loads(prompt)
unique_id = current_item
inputs = prompt[unique_id]['inputs']
will_execute = []
@ -213,6 +217,10 @@ def recursive_will_execute(prompt, outputs, current_item):
return will_execute + [unique_id]
def recursive_output_delete_if_changed(prompt, old_prompt, outputs, current_item):
if type(prompt) is not dict:
prompt = json.loads(prompt)
unique_id = current_item
inputs = prompt[unique_id]['inputs']
class_type = prompt[unique_id]['class_type']
@ -316,6 +324,10 @@ class PromptExecutor:
del d
def execute(self, prompt, prompt_id, extra_data={}, execute_outputs=[]):
if type(prompt) is not dict:
prompt = json.loads(prompt)
nodes.interrupt_processing(False)
if "client_id" in extra_data:
@ -594,6 +606,10 @@ def full_type_name(klass):
def validate_prompt(prompt):
outputs = set()
if type(prompt) is not dict:
prompt = json.loads(prompt)
for x in prompt:
class_ = nodes.NODE_CLASS_MAPPINGS[prompt[x]['class_type']]
if hasattr(class_, 'OUTPUT_NODE') and class_.OUTPUT_NODE == True: