From 35fa520625b9fa6151541fb9ab674eba46ce2ca5 Mon Sep 17 00:00:00 2001 From: yiyixuxu Date: Wed, 30 Apr 2025 23:07:47 +0200 Subject: [PATCH] up --- src/diffusers/pipelines/modular_pipeline.py | 2 +- src/diffusers/pipelines/pipeline_utils.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/diffusers/pipelines/modular_pipeline.py b/src/diffusers/pipelines/modular_pipeline.py index 1f1784b186..636b543395 100644 --- a/src/diffusers/pipelines/modular_pipeline.py +++ b/src/diffusers/pipelines/modular_pipeline.py @@ -172,7 +172,7 @@ class ModularPipelineMixin: # Import components loader (it is model-specific class) loader_class_name = MODULAR_LOADER_MAPPING[self.model_name] - diffusers_module = importlib.import_module(self.__module__.split(".")[0]) + diffusers_module = importlib.import_module("diffusers") loader_class = getattr(diffusers_module, loader_class_name) # Create deep copies to avoid modifying the original specs diff --git a/src/diffusers/pipelines/pipeline_utils.py b/src/diffusers/pipelines/pipeline_utils.py index c27cd434cd..22b0baee2e 100644 --- a/src/diffusers/pipelines/pipeline_utils.py +++ b/src/diffusers/pipelines/pipeline_utils.py @@ -1917,9 +1917,10 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin): f"{'' if k.startswith('_') else '_'}{k}": v for k, v in original_config.items() if k not in pipeline_kwargs } + optional_components = pipeline._optional_components if hasattr(pipeline, "_optional_components") and pipeline._optional_components else [] missing_modules = ( set(expected_modules) - - set(pipeline._optional_components) + - set(optional_components) - set(pipeline_kwargs.keys()) - set(true_optional_modules) )