diff --git a/src/diffusers/modular_pipelines/modular_pipeline.py b/src/diffusers/modular_pipelines/modular_pipeline.py index 8d85d60cd1..76a850b63c 100644 --- a/src/diffusers/modular_pipelines/modular_pipeline.py +++ b/src/diffusers/modular_pipelines/modular_pipeline.py @@ -724,7 +724,7 @@ class ConditionalPipelineBlocks(ModularPipelineBlocks): logger.error(error_msg) raise - def get_execution_blocks(self, **kwargs) -> "ModularPipelineBlocks" | None: + def get_execution_blocks(self, **kwargs) -> ModularPipelineBlocks | None: """ Get the block(s) that would execute given the inputs.