"tests/compute/test_batched_graph.py" did not exist on "c42eac718fe09aae40b7f8dfd8c3466b08e0f0be"
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):
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."
)
......
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