Skip to content

Commit 0a5c90e

Browse files
committed
add names property to pipeline blocks
1 parent 0fa5812 commit 0a5c90e

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed

src/diffusers/modular_pipelines/modular_pipeline.py

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -478,6 +478,23 @@ def combine_outputs(*named_output_lists: List[Tuple[str, List[OutputParam]]]) ->
478478
combined_dict[output_param.name] = output_param
479479

480480
return list(combined_dict.values())
481+
482+
@property
483+
def input_names(self) -> List[str]:
484+
return [input_param.name for input_param in self.inputs]
485+
486+
@property
487+
def intermediate_input_names(self) -> List[str]:
488+
return [input_param.name for input_param in self.intermediate_inputs]
489+
490+
@property
491+
def intermediate_output_names(self) -> List[str]:
492+
return [output_param.name for output_param in self.intermediate_outputs]
493+
494+
@property
495+
def output_names(self) -> List[str]:
496+
return [output_param.name for output_param in self.outputs]
497+
481498

482499

483500
class PipelineBlock(ModularPipelineBlocks):
@@ -2825,3 +2842,9 @@ def _dict_to_component_spec(
28252842
type_hint=type_hint,
28262843
**spec_dict,
28272844
)
2845+
2846+
2847+
def set_progress_bar_config(self, **kwargs):
2848+
for sub_block_name, sub_block in self.blocks.sub_blocks.items():
2849+
if hasattr(sub_block, "set_progress_bar_config"):
2850+
sub_block.set_progress_bar_config(**kwargs)

0 commit comments

Comments
 (0)