diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-28 05:30:17 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-28 05:30:17 +0000 |
| commit | 4c52dfe4ac98c53431ecd267d59f27391d3a63e7 (patch) | |
| tree | 6da0bb38b630a1d6628b55b722ab1a3f69daa72d /modules | |
| parent | 41975c375cf7167d1ac2e25caef2cb4da4ba4b31 (diff) | |
| download | stable-diffusion-webui-gfx803-4c52dfe4ac98c53431ecd267d59f27391d3a63e7.tar.gz stable-diffusion-webui-gfx803-4c52dfe4ac98c53431ecd267d59f27391d3a63e7.tar.bz2 stable-diffusion-webui-gfx803-4c52dfe4ac98c53431ecd267d59f27391d3a63e7.zip | |
make the detection for -v models less broad
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/sd_models_config.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_models_config.py b/modules/sd_models_config.py index 73854a45..00217990 100644 --- 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
|