make the detection for -v models less broad

2023-01-28 08:30:17 +03:00 · 2023-01-28 08:30:17 +03:00 · 4c52dfe4ac
commit 4c52dfe4ac
parent 41975c375c
1 changed files with 1 additions and 1 deletions
--- a/modules/sd_models_config.py
+++ b/modules/sd_models_config.py
@ -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