aboutsummaryrefslogtreecommitdiff
path: root/modules/sd_hijack.py
diff options
context:
space:
mode:
authorsuperhero-7 <537093830@qq.com>2023-10-01 12:25:19 +0800
committersuperhero-7 <537093830@qq.com>2023-10-01 12:25:19 +0800
commit2d947175b902d6838c803036d9757e7d3226b41d (patch)
treed62d9f599c94734d5e40b5ed711be13e0d4e081d /modules/sd_hijack.py
parentf8f4ff2bb8f56877dede466934dd8ddf25c21063 (diff)
fix linter issues
Diffstat (limited to 'modules/sd_hijack.py')
-rw-r--r--modules/sd_hijack.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py
index 4b36c0e9..0689699c 100644
--- a/modules/sd_hijack.py
+++ b/modules/sd_hijack.py
@@ -212,7 +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)
@@ -258,7 +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