michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 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 diffusion_model_input.shape[1] == 9:
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
else:
return config_sd2