wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-27 06:40:16 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

crash fix

Browse Source
This commit is contained in:
Dr.Lt.Data 2023-06-19 21:05:21 +09:00
parent 7859fb4f99
commit be62255663
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
worklist_execution.py
View File

@ -359,7 +359,7 @@ def worklist_output_delete_if_changed(prompt, old_prompt, outputs, next_nodes, m
if not to_delete: if not to_delete:
if is_changed != is_changed_old: if is_changed != is_changed_old:
to_delete = True to_delete = True
elif unique_id in muted_nodes or has_muted_input(old_prompt[unique_id]['inputs']): elif unique_id in muted_nodes or (unique_id in old_prompt and has_muted_input(old_prompt[unique_id]['inputs'])):
continue continue
elif unique_id not in old_prompt: elif unique_id not in old_prompt:
to_delete = True to_delete = True
@ -453,8 +453,12 @@ 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)
muted_nodes = set([str(node['id']) for node in extra_data['extra_pnginfo']['workflow']['nodes']]) - set(prompt.keys()) if 'extra_pnginfo' in extra_data:
unmuted_nodes = self.prev_muted_nodes - muted_nodes muted_nodes = set([str(node['id']) for node in extra_data['extra_pnginfo']['workflow']['nodes']]) - set(prompt.keys())
unmuted_nodes = self.prev_muted_nodes - muted_nodes
else:
muted_nodes = set()
unmuted_nodes = set()
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"]