Browse Source

Print error when node is missing.

pull/3493/merge
comfyanonymous 6 months ago
parent
commit
276f8fce9f
  1. 14
      execution.py

14
execution.py

@ -625,13 +625,23 @@ def validate_prompt(prompt):
if 'class_type' not in prompt[x]: if 'class_type' not in prompt[x]:
error = { error = {
"type": "invalid_prompt", "type": "invalid_prompt",
"message": f"Cannot execute due to a missing node", "message": f"Cannot execute because a node is missing the class_type property.",
"details": f"Node ID '#{x}'",
"extra_info": {}
}
return (False, error, [], [])
class_type = prompt[x]['class_type']
class_ = nodes.NODE_CLASS_MAPPINGS.get(class_type, None)
if class_ is None:
error = {
"type": "invalid_prompt",
"message": f"Cannot execute because node {class_type} does not exist.",
"details": f"Node ID '#{x}'", "details": f"Node ID '#{x}'",
"extra_info": {} "extra_info": {}
} }
return (False, error, [], []) return (False, error, [], [])
class_ = nodes.NODE_CLASS_MAPPINGS[prompt[x]['class_type']]
if hasattr(class_, 'OUTPUT_NODE') and class_.OUTPUT_NODE is True: if hasattr(class_, 'OUTPUT_NODE') and class_.OUTPUT_NODE is True:
outputs.add(x) outputs.add(x)

Loading…
Cancel
Save