qnloft
/
ComfyUI
mirror of https://github.com/comfyanonymous/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Improve execution UX. 

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 fca42836f2
commit 14af129c55
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):