aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_hijack.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-02-19 11:15:41 +0300
committerGitHub <noreply@github.com>2023-02-19 11:15:41 +0300
commit226bc0465373b55ca8b5fb18e9277c04974a04a8 (patch)
tree826cb97564051353c1c2d5a591651be034f404db /modules/sd_hijack.py
parent3fcc0873173b3ffbeafde766332ff814969e1724 (diff)
parent2016733814433ca2b69d10764bfa0ab4c7088782 (diff)
downloadstable-diffusion-webui-gfx803-226bc0465373b55ca8b5fb18e9277c04974a04a8.tar.gz
Merge pull request #7637 from brkirch/fix-hypernetworks-pix2pix
Fix hypernetworks and instruct pix2pix not working with `--upcast-sampling`
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