Unverified Commit c28db0aa authored by hlky's avatar hlky Committed by GitHub
Browse files

Fix AutoPipeline `from_pipe` where source pipeline is missing target...


Fix AutoPipeline `from_pipe` where source pipeline is missing target pipeline's optional components (#10400)

* Optional components in AutoPipeline

* missing_modules

---------
Co-authored-by: default avatarYiYi Xu <yixu310@gmail.com>
parent f7822ae4
...@@ -528,7 +528,9 @@ class AutoPipelineForText2Image(ConfigMixin): ...@@ -528,7 +528,9 @@ class AutoPipelineForText2Image(ConfigMixin):
if k not in text_2_image_kwargs if k not in text_2_image_kwargs
} }
missing_modules = set(expected_modules) - set(pipeline._optional_components) - set(text_2_image_kwargs.keys()) missing_modules = (
set(expected_modules) - set(text_2_image_cls._optional_components) - set(text_2_image_kwargs.keys())
)
if len(missing_modules) > 0: if len(missing_modules) > 0:
raise ValueError( raise ValueError(
...@@ -838,7 +840,9 @@ class AutoPipelineForImage2Image(ConfigMixin): ...@@ -838,7 +840,9 @@ class AutoPipelineForImage2Image(ConfigMixin):
if k not in image_2_image_kwargs if k not in image_2_image_kwargs
} }
missing_modules = set(expected_modules) - set(pipeline._optional_components) - set(image_2_image_kwargs.keys()) missing_modules = (
set(expected_modules) - set(image_2_image_cls._optional_components) - set(image_2_image_kwargs.keys())
)
if len(missing_modules) > 0: if len(missing_modules) > 0:
raise ValueError( raise ValueError(
...@@ -1141,7 +1145,9 @@ class AutoPipelineForInpainting(ConfigMixin): ...@@ -1141,7 +1145,9 @@ class AutoPipelineForInpainting(ConfigMixin):
if k not in inpainting_kwargs if k not in inpainting_kwargs
} }
missing_modules = set(expected_modules) - set(pipeline._optional_components) - set(inpainting_kwargs.keys()) missing_modules = (
set(expected_modules) - set(inpainting_cls._optional_components) - set(inpainting_kwargs.keys())
)
if len(missing_modules) > 0: if len(missing_modules) > 0:
raise ValueError( raise ValueError(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment