From bb46be2f18c09aaefefd10f53c804c9ea604786f Mon Sep 17 00:00:00 2001 From: Aryan V S Date: Tue, 31 Oct 2023 00:02:11 +0530 Subject: [PATCH] Fix incorrect loading of custom pipeline (#5568) * update * update * update * update --- src/diffusers/pipelines/pipeline_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/pipeline_utils.py b/src/diffusers/pipelines/pipeline_utils.py index 512cf8d567..8baafbaef1 100644 --- a/src/diffusers/pipelines/pipeline_utils.py +++ b/src/diffusers/pipelines/pipeline_utils.py @@ -1669,7 +1669,7 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin): for component in folder_names: module_candidate = config_dict[component][0] - if module_candidate is None: + if module_candidate is None or not isinstance(module_candidate, str): continue candidate_file = os.path.join(component, module_candidate + ".py")