wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-10 05:22:34 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

robust patch

Browse Source
This commit is contained in:
Dr.Lt.Data 2023-07-17 00:55:13 +09:00
parent eb1ff03da0
commit ef5680be7b
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
execution.py
View File

@ -64,7 +64,7 @@ def map_node_over_list(obj, input_data_all, func, allow_interrupt=False):
for k, v in d.items(): for k, v in d.items():
if not v: if not v:
return None continue
else: else:
d_new[k] = v[i if len(v) > i else -1] d_new[k] = v[i if len(v) > i else -1]
return d_new return d_new

4
worklist_execution.py
View File

@ -90,7 +90,7 @@ def is_incomplete_input_slots(class_def, inputs, outputs):
if isinstance(input_data, list): if isinstance(input_data, list):
input_unique_id = input_data[0] input_unique_id = input_data[0]
if input_unique_id in outputs and outputs[input_unique_id][input_data[1]] != [None]: if input_unique_id in outputs and len(outputs[input_unique_id]) > 0 and outputs[input_unique_id][input_data[1]] != [None]:
return False return False
return True return True
@ -472,7 +472,7 @@ class PromptExecutor:
to_delete += [o] to_delete += [o]
else: else:
p = prompt[o[0]] p = prompt[o[0]]
if o[1] != p['class_type']: if ('class_type' in p and o[1] != p['class_type']) or ('class_type' not in p and o[1] != 'DummyNode'):
to_delete += [o] to_delete += [o]
for o in to_delete: for o in to_delete:
d = self.object_storage.pop(o) d = self.object_storage.pop(o)