xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-16 04:43:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Improve execution UX.

Browse Source 
Some branches with VAELoader -> VAEDecode -> Preview were being executed
last. With this change they will be executed earlier.
This commit is contained in:
comfyanonymous 2024-08-17 11:36:10 -04:00
parent 03c874ec6e
commit 5c270c22e1
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_execution/graph.py
View File

@ -202,8 +202,14 @@ class ExecutionList(TopologicalSort):
if is_output(blocked_node_id):
return node_id
#Do we want to look deeper?
#This should handle the VAELoader -> VAEDecode -> preview case
for node_id in node_list:
for blocked_node_id in self.blocking[node_id]:
for blocked_node_id1 in self.blocking[blocked_node_id]:
if is_output(blocked_node_id1):
return node_id
#TODO: this function should be improved
return node_list[0]
def unstage_node_execution(self):