diff options
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 57ed5635..79476783 100644 --- a/modules/sd_hijack.py +++ b/modules/sd_hijack.py @@ -154,6 +154,8 @@ class StableDiffusionModelHijack: m.cond_stage_model = sd_hijack_open_clip.FrozenOpenCLIPEmbedderWithCustomWords(m.cond_stage_model, self)
apply_weighted_forward(m)
+ if m.cond_stage_key == "edit":
+ sd_hijack_unet.hijack_ddpm_edit()
self.optimization_method = apply_optimizations()
|