Skip to content

Commit d131449

Browse files
committed
fix empty collector after grouped materialization
1 parent d0a0e7e commit d131449

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

invokeai/app/services/shared/graph.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -698,9 +698,12 @@ def prepare(self, base_g: Optional[nx.DiGraph] = None) -> Optional[str]:
698698
new_node_ids: list[str] = []
699699

700700
if isinstance(next_node, CollectInvocation):
701-
for iteration_path, iteration_mappings in self._get_collect_iteration_mapping_groups(
701+
iteration_mapping_groups = self._get_collect_iteration_mapping_groups(
702702
self._state.graph._get_input_edges(next_node_id)
703-
):
703+
)
704+
if not iteration_mapping_groups:
705+
iteration_mapping_groups = [((), [])]
706+
for iteration_path, iteration_mappings in iteration_mapping_groups:
704707
create_results = self.create_execution_node(next_node_id, iteration_mappings, iteration_path)
705708
if create_results is not None:
706709
new_node_ids.extend(create_results)

0 commit comments

Comments
 (0)