From 46442f90a2ce70c1c706a69c9ab8154e719214b3 Mon Sep 17 00:00:00 2001 From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com> Date: Mon, 5 Aug 2024 03:17:35 -0700 Subject: [PATCH] Update loader.py --- backend/loader.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backend/loader.py b/backend/loader.py index 8072299a..1a26ff2b 100644 --- a/backend/loader.py +++ b/backend/loader.py @@ -23,7 +23,7 @@ logging.getLogger("diffusers").setLevel(logging.ERROR) dir_path = os.path.dirname(__file__) -def load_component(guess, component_name, lib_name, cls_name, repo_path, state_dict): +def load_huggingface_component(guess, component_name, lib_name, cls_name, repo_path, state_dict): config_path = os.path.join(repo_path, component_name) if component_name in ['feature_extractor', 'safety_checker']: @@ -103,7 +103,7 @@ def forge_loader(sd): if isinstance(v, list) and len(v) == 2: lib_name, cls_name = v component_sd = state_dicts.get(component_name, None) - component = load_component(estimated_config, component_name, lib_name, cls_name, local_path, component_sd) + component = load_huggingface_component(estimated_config, component_name, lib_name, cls_name, local_path, component_sd) if component_sd is not None: del state_dicts[component_name] if component is not None: