fix: for checking mandatory and optional pipeline components (#11189)

elismasilva · web-flow · commit be0b7f55cc32 · 2025-04-02T08:07:24.000+01:00
fix: optional componentes verification on load
diff --git a/src/diffusers/pipelines/pipeline_utils.py b/src/diffusers/pipelines/pipeline_utils.py
@@ -998,7 +998,7 @@ def load_module(name, value):
             for module in missing_modules:
                 init_kwargs[module] = passed_class_obj.get(module, None)
         elif len(missing_modules) > 0:
-            passed_modules = set(list(init_kwargs.keys()) + list(passed_class_obj.keys())) - optional_kwargs
+            passed_modules = set(list(init_kwargs.keys()) + list(passed_class_obj.keys())) - set(optional_kwargs)
             raise ValueError(
                 f"Pipeline {pipeline_class} expected {expected_modules}, but only {passed_modules} were passed."
             )

Original file line number	Diff line number	Diff line change
`@@ -998,7 +998,7 @@ def load_module(name, value):`
`998`	`998`	`for module in missing_modules:`
`999`	`999`	`init_kwargs[module] = passed_class_obj.get(module, None)`
`1000`	`1000`	`elif len(missing_modules) > 0:`
`1001`		`- passed_modules = set(list(init_kwargs.keys()) + list(passed_class_obj.keys())) - optional_kwargs`
	`1001`	`+ passed_modules = set(list(init_kwargs.keys()) + list(passed_class_obj.keys())) - set(optional_kwargs)`
`1002`	`1002`	`raise ValueError(`
`1003`	`1003`	`f"Pipeline {pipeline_class} expected {expected_modules}, but only {passed_modules} were passed."`
`1004`	`1004`	`)`