From 5714f79301e95f7c535e7167304467b56ce9a347 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 05:18:53 -0800 Subject: [PATCH] Update sd_hijack.py --- modules/sd_hijack.py | 12 +----------- 1 file changed, 1 insertion(+), 11 deletions(-) diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py index 7917d1a8..f048e990 100644 --- a/modules/sd_hijack.py +++ b/modules/sd_hijack.py @@ -192,17 +192,7 @@ class StableDiffusionModelHijack: return res self.layers = flatten(m) - - import modules.models.diffusion.ddpm_edit - - if isinstance(m, ldm.models.diffusion.ddpm.LatentDiffusion): - sd_unet.original_forward = ldm_original_forward - elif isinstance(m, modules.models.diffusion.ddpm_edit.LatentDiffusion): - sd_unet.original_forward = ldm_original_forward - elif isinstance(m, sgm.models.diffusion.DiffusionEngine): - sd_unet.original_forward = sgm_original_forward - else: - sd_unet.original_forward = None + sd_unet.original_forward = None def undo_hijack(self, m): pass