Unverified Commit be0b7f55 authored by Eliseu Silva's avatar Eliseu Silva Committed by GitHub
Browse files

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

fix: optional componentes verification on load
parent 4d5a96e4
...@@ -998,7 +998,7 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin): ...@@ -998,7 +998,7 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin):
for module in missing_modules: for module in missing_modules:
init_kwargs[module] = passed_class_obj.get(module, None) init_kwargs[module] = passed_class_obj.get(module, None)
elif len(missing_modules) > 0: 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( raise ValueError(
f"Pipeline {pipeline_class} expected {expected_modules}, but only {passed_modules} were passed." f"Pipeline {pipeline_class} expected {expected_modules}, but only {passed_modules} were passed."
) )
......
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