From 9c1c0da026cb7ef091a0f3fa24b14ae8634f6de5 Mon Sep 17 00:00:00 2001
From: AUTOMATIC1111 <16777216c@gmail.com>
Date: Mon, 6 Nov 2023 11:17:36 +0300
Subject: [PATCH] fix exception related to the pix2pix

---
 modules/sd_hijack.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py
index fba23c38b..0157e19f0 100644
--- a/modules/sd_hijack.py
+++ b/modules/sd_hijack.py
@@ -256,8 +256,12 @@ class StableDiffusionModelHijack:
 
         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: