lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-29 02:31:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

make model guess a function that can be patched

Browse Source
This commit is contained in:
layerdiffusion
2024-07-30 17:26:49 -06:00
parent c8156fcf41
commit f052fabd4d
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
backend/loader.py
View File

@@ -23,9 +23,14 @@ def load_component(component_name, lib_name, cls_name, repo_path, sd):
return None return None
def guess_repo_name_from_state_dict(sd):
result = fetch_diffusers_config(sd)['pretrained_model_name_or_path']
return result
def load_huggingface_components(sd): def load_huggingface_components(sd):
pretrained_model_name_or_path = fetch_diffusers_config(sd)['pretrained_model_name_or_path'] repo_name = guess_repo_name_from_state_dict(sd)
local_path = os.path.join(dir_path, 'huggingface', pretrained_model_name_or_path) local_path = os.path.join(dir_path, 'huggingface', repo_name)
config = DiffusionPipeline.load_config(local_path) config = DiffusionPipeline.load_config(local_path)
result = {"repo_path": local_path} result = {"repo_path": local_path}
for component_name, v in config.items(): for component_name, v in config.items():