Commit 4c52dfe4 authored by AUTOMATIC's avatar AUTOMATIC

make the detection for -v models less broad

parent 41975c37
...@@ -31,7 +31,7 @@ def guess_model_config_from_state_dict(sd, filename): ...@@ -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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment