mrq/stable-diffusion-webui
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

make the detection for -v models less broad

Browse Source
This commit is contained in:
AUTOMATIC 2023-01-28 08:30:17 +03:00
parent 41975c375c
commit 4c52dfe4ac
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_models_config.py
View File

@ -31,7 +31,7 @@ def guess_model_config_from_state_dict(sd, filename):
if sd2_cond_proj_weight is not None and sd2_cond_proj_weight.shape[1] == 1024: if sd2_cond_proj_weight is not None and sd2_cond_proj_weight.shape[1] == 1024:
if diffusion_model_input.shape[1] == 9: if diffusion_model_input.shape[1] == 9:
return config_sd2_inpainting return config_sd2_inpainting
elif re.search(re_parametrization_v, fn) or "768" in fn: elif re.search(re_parametrization_v, fn):
return config_sd2v return config_sd2v
else: else:
return config_sd2 return config_sd2