support altdiffusion-m18

1 files changed, 2 insertions, 0 deletions

diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py
index ae9b2a65..4b36c0e9 100644
--- a/modules/sd_hijack.py
+++ b/modules/sd_hijack.py
@@ -212,6 +212,7 @@ class StableDiffusionModelHijack:
             model_embeddings = m.cond_stage_model.roberta.embeddings

             model_embeddings.token_embedding = EmbeddingsWithFixes(model_embeddings.word_embeddings, self)

             m.cond_stage_model = sd_hijack_xlmr.FrozenXLMREmbedderWithCustomWords(m.cond_stage_model, self)

+            

         elif type(m.cond_stage_model) == ldm.modules.encoders.modules.FrozenCLIPEmbedder:

             model_embeddings = m.cond_stage_model.transformer.text_model.embeddings

             model_embeddings.token_embedding = EmbeddingsWithFixes(model_embeddings.token_embedding, self)

@@ -257,6 +258,7 @@ class StableDiffusionModelHijack:
 

             if hasattr(m, 'cond_stage_model'):

                 delattr(m, 'cond_stage_model')

+                

         elif type(m.cond_stage_model) == sd_hijack_xlmr.FrozenXLMREmbedderWithCustomWords:

             m.cond_stage_model = m.cond_stage_model.wrapped