AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

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

Browse Source 
* Optional components in AutoPipeline

* missing_modules

---------

Co-authored-by: YiYi Xu <yixu310@gmail.com>
This commit is contained in:
hlky
2025-01-02 21:06:51 +00:00
committed by GitHub
parent f7822ae4bf
commit c28db0aa5b
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/diffusers/pipelines/auto_pipeline.py
View File

@@ -528,7 +528,9 @@ class AutoPipelineForText2Image(ConfigMixin):
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:
raise ValueError(
@@ -838,7 +840,9 @@ class AutoPipelineForImage2Image(ConfigMixin):
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:
raise ValueError(
@@ -1141,7 +1145,9 @@ class AutoPipelineForInpainting(ConfigMixin):
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:
raise ValueError(