Skip to content
Merged
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions src/diffusers/modular_pipelines/modular_pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -861,6 +861,10 @@ def __init__(self):
else:
sub_blocks[block_name] = block
self.sub_blocks = sub_blocks
if not len(self.block_names) == len(self.block_classes):
Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

adding a check here, we had a similar check to autopipeline blocks, which I got this error message a couple times already so it's helpful

raise ValueError(
f"In {self.__class__.__name__}, the number of block_names and block_classes must be the same."
)

def _get_inputs(self):
inputs = []
Expand Down
Loading