aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_hijack.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-02-19 09:30:58 +0000
committerGitHub <noreply@github.com>2023-02-19 09:30:58 +0000
commitcfc9849f3f64977936769b6479d6b2231ecbfc5b (patch)
tree51b29ea3b255fe43b0fe7560f9f7fdd23a475427 /modules/sd_hijack.py
parent5afd9e82c3829348c58803cd85b02c87308fffae (diff)
parentd99bd04b3f8c7753e31aa6dea6109785c4bb92c9 (diff)
downloadstable-diffusion-webui-gfx803-cfc9849f3f64977936769b6479d6b2231ecbfc5b.tar.gz
stable-diffusion-webui-gfx803-cfc9849f3f64977936769b6479d6b2231ecbfc5b.tar.bz2
stable-diffusion-webui-gfx803-cfc9849f3f64977936769b6479d6b2231ecbfc5b.zip
Merge branch 'master' into 6866-fix-hires-prompt-matrix
Diffstat (limited to 'modules/sd_hijack.py')
-rw-r--r--modules/sd_hijack.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py
index 8fdc5990..fca418cd 100644
--- a/modules/sd_hijack.py
+++ b/modules/sd_hijack.py
@@ -104,6 +104,9 @@ class StableDiffusionModelHijack:
m.cond_stage_model.model.token_embedding = EmbeddingsWithFixes(m.cond_stage_model.model.token_embedding, self)
m.cond_stage_model = sd_hijack_open_clip.FrozenOpenCLIPEmbedderWithCustomWords(m.cond_stage_model, self)
+ if m.cond_stage_key == "edit":
+ sd_hijack_unet.hijack_ddpm_edit()
+
self.optimization_method = apply_optimizations()
self.clip = m.cond_stage_model