lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-23 07:49:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update loader.py

Browse Source
This commit is contained in:
layerdiffusion
2024-08-05 03:17:35 -07:00
parent 4189045323
commit 46442f90a2
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
backend/loader.py
View File

@@ -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: