Skip to content

Commit c28db0a

Browse files
hlkyyiyixuxu
andauthored
Fix AutoPipeline from_pipe where source pipeline is missing target pipeline's optional components (#10400)
* Optional components in AutoPipeline * missing_modules --------- Co-authored-by: YiYi Xu <[email protected]>
1 parent f7822ae commit c28db0a

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

src/diffusers/pipelines/auto_pipeline.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -528,7 +528,9 @@ def from_pipe(cls, pipeline, **kwargs):
528528
if k not in text_2_image_kwargs
529529
}
530530

531-
missing_modules = set(expected_modules) - set(pipeline._optional_components) - set(text_2_image_kwargs.keys())
531+
missing_modules = (
532+
set(expected_modules) - set(text_2_image_cls._optional_components) - set(text_2_image_kwargs.keys())
533+
)
532534

533535
if len(missing_modules) > 0:
534536
raise ValueError(
@@ -838,7 +840,9 @@ def from_pipe(cls, pipeline, **kwargs):
838840
if k not in image_2_image_kwargs
839841
}
840842

841-
missing_modules = set(expected_modules) - set(pipeline._optional_components) - set(image_2_image_kwargs.keys())
843+
missing_modules = (
844+
set(expected_modules) - set(image_2_image_cls._optional_components) - set(image_2_image_kwargs.keys())
845+
)
842846

843847
if len(missing_modules) > 0:
844848
raise ValueError(
@@ -1141,7 +1145,9 @@ def from_pipe(cls, pipeline, **kwargs):
11411145
if k not in inpainting_kwargs
11421146
}
11431147

1144-
missing_modules = set(expected_modules) - set(pipeline._optional_components) - set(inpainting_kwargs.keys())
1148+
missing_modules = (
1149+
set(expected_modules) - set(inpainting_cls._optional_components) - set(inpainting_kwargs.keys())
1150+
)
11451151

11461152
if len(missing_modules) > 0:
11471153
raise ValueError(

0 commit comments

Comments
 (0)