diff options
| author | superhero-7 <537093830@qq.com> | 2023-09-23 09:55:19 +0000 |
|---|---|---|
| committer | superhero-7 <537093830@qq.com> | 2023-09-23 09:55:19 +0000 |
| commit | f8f4ff2bb8f56877dede466934dd8ddf25c21063 (patch) | |
| tree | 43e2635cc733e4580d7a14c2de915211827f462d /modules/sd_hijack.py | |
| parent | 702a1e1cc70240f2adbcfb707a644a5a98b5443c (diff) | |
| download | stable-diffusion-webui-gfx803-f8f4ff2bb8f56877dede466934dd8ddf25c21063.tar.gz stable-diffusion-webui-gfx803-f8f4ff2bb8f56877dede466934dd8ddf25c21063.tar.bz2 stable-diffusion-webui-gfx803-f8f4ff2bb8f56877dede466934dd8ddf25c21063.zip | |
support altdiffusion-m18
Diffstat (limited to 'modules/sd_hijack.py')
| -rw-r--r-- | modules/sd_hijack.py | 2 |
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
|