wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-26 14:20:27 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

bugfix: loop explosion by multiple input nodes

Browse Source
This commit is contained in:
Dr.Lt.Data 2023-06-17 20:05:57 +09:00
parent d60b306766
commit 8b0ec819ff
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
worklist_execution.py
View File

@ -156,6 +156,10 @@ def worklist_execute(server, prompt, outputs, extra_data, prompt_id, outputs_ui,
def add_work(item): def add_work(item):
worklist.put(item) worklist.put(item)
cnt = will_execute.get(item, 0) cnt = will_execute.get(item, 0)
if item in outputs:
del outputs[item]
will_execute[item] = cnt + 1 will_execute[item] = cnt + 1
def get_work(): def get_work():
@ -227,6 +231,9 @@ def worklist_execute(server, prompt, outputs, extra_data, prompt_id, outputs_ui,
nonlocal input_data_all nonlocal input_data_all
input_data_all = get_input_data(inputs, class_def, unique_id, outputs, prompt, extra_data) input_data_all = get_input_data(inputs, class_def, unique_id, outputs, prompt, extra_data)
if input_data_all is None:
return
if server.client_id is not None: if server.client_id is not None:
server.last_node_id = unique_id server.last_node_id = unique_id
server.send_sync("executing", {"node": unique_id, "prompt_id": prompt_id, "progress": get_progress()}, server.send_sync("executing", {"node": unique_id, "prompt_id": prompt_id, "progress": get_progress()},
@ -244,6 +251,10 @@ def worklist_execute(server, prompt, outputs, extra_data, prompt_id, outputs_ui,
executed.add(unique_id) executed.add(unique_id)
result = exception_helper(unique_id, input_data_all, executed, outputs, task) result = exception_helper(unique_id, input_data_all, executed, outputs, task)
if input_data_all is None:
continue
if result is not None: if result is not None:
return result # error state return result # error state
else: else: