298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-06-01 17:18:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove all fields not defined in nodes from prompt before execution.

Browse Source
This commit is contained in:
comfyanonymous 2025-05-28 00:04:08 -04:00
parent 06c661004e
commit ba55063ac5
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
execution.py
View File

@ -435,6 +435,20 @@ def execute(server, dynprompt, caches, current_item, extra_data, executed, promp
return (ExecutionResult.SUCCESS, None, None) return (ExecutionResult.SUCCESS, None, None)
def clean_inputs(prompt):
for unique_id, node in prompt.items():
inputs = node['inputs']
class_type = node['class_type']
obj_class = nodes.NODE_CLASS_MAPPINGS[class_type]
class_inputs = obj_class.INPUT_TYPES()
valid_inputs = set(class_inputs.get('required',{})).union(set(class_inputs.get('optional',{})))
for k in list(inputs.keys()):
if k not in valid_inputs:
inputs.pop(k)
return prompt
class PromptExecutor: class PromptExecutor:
def __init__(self, server, cache_type=False, cache_size=None): def __init__(self, server, cache_type=False, cache_size=None):
self.cache_size = cache_size self.cache_size = cache_size
@ -486,6 +500,7 @@ class PromptExecutor:
def execute(self, prompt, prompt_id, extra_data={}, execute_outputs=[]): def execute(self, prompt, prompt_id, extra_data={}, execute_outputs=[]):
nodes.interrupt_processing(False) nodes.interrupt_processing(False)
prompt = clean_inputs(prompt)
if "client_id" in extra_data: if "client_id" in extra_data:
self.server.client_id = extra_data["client_id"] self.server.client_id = extra_data["client_id"]