From 0eb7f285020dfe90ad167ef15244e32a05e91ce8 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 05:07:44 -0800 Subject: [PATCH] Update sd_hijack.py --- modules/sd_hijack.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py index f5a5a696..5cbaf908 100644 --- a/modules/sd_hijack.py +++ b/modules/sd_hijack.py @@ -180,12 +180,10 @@ class StableDiffusionModelHijack: model_embeddings.token_embedding = EmbeddingsWithFixes(model_embeddings.token_embedding, self) 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.clip = m.cond_stage_model + apply_weighted_forward(m) + def flatten(el): flattened = [flatten(children) for children in el.children()] res = [el]