up

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

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)
         )