From f01a78d24a2b198cb13b1aea1eeb3a1ef3d797c7 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 05:13:46 -0800 Subject: [PATCH] Update sd_hijack.py --- modules/sd_hijack.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py index 5cbaf908..7917d1a8 100644 --- a/modules/sd_hijack.py +++ b/modules/sd_hijack.py @@ -152,7 +152,7 @@ class StableDiffusionModelHijack: if typename == 'FrozenOpenCLIPEmbedder': model_embeddings = embedder.transformer.text_model.embeddings model_embeddings.token_embedding = EmbeddingsWithFixes(model_embeddings.token_embedding, self) - conditioner.embedders[i] = sd_hijack_open_clip.FrozenOpenCLIPEmbedderWithCustomWords(embedder, self) + conditioner.embedders[i] = sd_hijack_clip.FrozenCLIPEmbedderForSDXLWithCustomWords(embedder, self) text_cond_models.append(conditioner.embedders[i]) if typename == 'FrozenCLIPEmbedder': model_embeddings = embedder.transformer.text_model.embeddings @@ -162,7 +162,7 @@ class StableDiffusionModelHijack: if typename == 'FrozenOpenCLIPEmbedder2': model_embeddings = embedder.transformer.text_model.embeddings model_embeddings.token_embedding = EmbeddingsWithFixes(model_embeddings.token_embedding, self, textual_inversion_key='clip_g') - conditioner.embedders[i] = sd_hijack_open_clip.FrozenOpenCLIPEmbedder2WithCustomWords(embedder, self) + conditioner.embedders[i] = sd_hijack_clip.FrozenCLIPEmbedderForSDXLWithCustomWords(embedder, self) text_cond_models.append(conditioner.embedders[i]) if len(text_cond_models) == 1: @@ -178,7 +178,7 @@ class StableDiffusionModelHijack: elif type(m.cond_stage_model) == ldm.modules.encoders.modules.FrozenOpenCLIPEmbedder: model_embeddings = m.cond_stage_model.transformer.text_model.embeddings model_embeddings.token_embedding = EmbeddingsWithFixes(model_embeddings.token_embedding, self) - m.cond_stage_model = sd_hijack_open_clip.FrozenOpenCLIPEmbedderWithCustomWords(m.cond_stage_model, self) + m.cond_stage_model = sd_hijack_clip.FrozenCLIPEmbedderWithCustomWords(m.cond_stage_model, self) self.clip = m.cond_stage_model