Merge branch 'release_candidate'

1 files changed, 4 insertions, 1 deletions

diff --git a/modules/sd_models_config.py b/modules/sd_models_config.py
index 08dd03f1..deab2f6e 100644
--- a/modules/sd_models_config.py
+++ b/modules/sd_models_config.py
@@ -21,7 +21,7 @@ config_unopenclip = os.path.join(sd_repo_configs_path, "v2-1-stable-unclip-h-inf
 config_inpainting = os.path.join(sd_configs_path, "v1-inpainting-inference.yaml")

 config_instruct_pix2pix = os.path.join(sd_configs_path, "instruct-pix2pix.yaml")

 config_alt_diffusion = os.path.join(sd_configs_path, "alt-diffusion-inference.yaml")

-

+config_alt_diffusion_m18 = os.path.join(sd_configs_path, "alt-diffusion-m18-inference.yaml")

 

 def is_using_v_parameterization_for_sd2(state_dict):

     """

@@ -95,7 +95,10 @@ def guess_model_config_from_state_dict(sd, filename):
         if diffusion_model_input.shape[1] == 8:

             return config_instruct_pix2pix

 

+

     if sd.get('cond_stage_model.roberta.embeddings.word_embeddings.weight', None) is not None:

+        if sd.get('cond_stage_model.transformation.weight').size()[0] == 1024:

+            return config_alt_diffusion_m18

         return config_alt_diffusion

 

     return config_default