Skip to content

GitLab

Unverified Commit bb46be2f authored by Aryan V S's avatar Aryan V S Committed by GitHub
Browse files

Fix incorrect loading of custom pipeline (#5568) 

* update

* update

* update

* update
parent ac7b1716
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/pipelines/pipeline_utils.py
View file @ bb46be2f
...@@ -1669,7 +1669,7 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin): ...@@ -1669,7 +1669,7 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin):
for component in folder_names: for component in folder_names:
module_candidate = config_dict[component][0] module_candidate = config_dict[component][0]
if module_candidate is None: if module_candidate is None or not isinstance(module_candidate, str):
continue continue
candidate_file = os.path.join(component, module_candidate + ".py") candidate_file = os.path.join(component, module_candidate + ".py")
......
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