| from dataclasses import replace | |
| from diffusers.modular_pipelines import ModularPipelineBlocks | |
| class ResolvableBlock(ModularPipelineBlocks): | |
| """Leaf block whose generic component specs can carry the model repository's resolved classes and location. | |
| `Pixal3DBlocks(components_repo=...)` fills `component_overrides`, so a sub-block popped out of the pipeline still | |
| loads the right custom classes from the right repository. | |
| """ | |
| component_overrides = {} | |
| def resolve(self, specs): | |
| return [ | |
| replace(spec, **self.component_overrides[spec.name]) if spec.name in self.component_overrides else spec | |
| for spec in specs | |
| ] | |