Skip to content

Commit

Permalink
List cached nodes in execution_cached message
Browse files Browse the repository at this point in the history
This was previously just bugged in this PR.
  • Loading branch information
guill committed Aug 8, 2024
1 parent 36131f0 commit fd7229e
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions execution.py
Original file line number Diff line number Diff line change
Expand Up @@ -466,15 +466,19 @@ def execute(self, prompt, prompt_id, extra_data={}, execute_outputs=[]):
cache.set_prompt(dynamic_prompt, prompt.keys(), is_changed_cache)
cache.clean_unused()

current_outputs = self.caches.outputs.all_node_ids()
cached_nodes = []
for node_id in prompt:
if self.caches.outputs.get(node_id) is not None:
cached_nodes.append(node_id)

comfy.model_management.cleanup_models(keep_clone_weights_loaded=True)
self.add_message("execution_cached",
{ "nodes": list(current_outputs) , "prompt_id": prompt_id},
{ "nodes": cached_nodes, "prompt_id": prompt_id},
broadcast=False)
pending_subgraph_results = {}
executed = set()
execution_list = ExecutionList(dynamic_prompt, self.caches.outputs)
current_outputs = self.caches.outputs.all_node_ids()
for node_id in list(execute_outputs):
execution_list.add_node(node_id)

Expand Down

0 comments on commit fd7229e

Please sign in to comment.