diff --git a/configs/instruct-pix2pix.yaml b/configs/instruct-pix2pix.yaml
index 4e896879d..76993f8fd 100644
--- a/configs/instruct-pix2pix.yaml
+++ b/configs/instruct-pix2pix.yaml
@@ -3,7 +3,7 @@
 
 model:
   base_learning_rate: 1.0e-04
-  target: modules.models.diffusion.ddpm_edit.LatentDiffusion
+  target: modules.hijack.ddpm_edit.LatentDiffusion
   params:
     linear_start: 0.00085
     linear_end: 0.0120
diff --git a/modules/models/diffusion/ddpm_edit.py b/modules/hijack/ddpm_edit.py
similarity index 100%
rename from modules/models/diffusion/ddpm_edit.py
rename to modules/hijack/ddpm_edit.py
diff --git a/modules/sd_hijack_unet.py b/modules/sd_hijack_unet.py
index 3234d33b6..ea8773147 100644
--- a/modules/sd_hijack_unet.py
+++ b/modules/sd_hijack_unet.py
@@ -59,9 +59,9 @@ ddpm_edit_hijack = None
 def hijack_ddpm_edit():
     global ddpm_edit_hijack # pylint: disable=global-statement
     if not ddpm_edit_hijack:
-        CondFunc('modules.models.diffusion.ddpm_edit.LatentDiffusion.decode_first_stage', first_stage_sub, first_stage_cond)
-        CondFunc('modules.models.diffusion.ddpm_edit.LatentDiffusion.encode_first_stage', first_stage_sub, first_stage_cond)
-        ddpm_edit_hijack = CondFunc('modules.models.diffusion.ddpm_edit.LatentDiffusion.apply_model', apply_model, unet_needs_upcast)
+        CondFunc('modules.hijack.ddpm_edit.LatentDiffusion.decode_first_stage', first_stage_sub, first_stage_cond)
+        CondFunc('modules.hijack.ddpm_edit.LatentDiffusion.encode_first_stage', first_stage_sub, first_stage_cond)
+        ddpm_edit_hijack = CondFunc('modules.hijack.ddpm_edit.LatentDiffusion.apply_model', apply_model, unet_needs_upcast)
 
 
 unet_needs_upcast = lambda *args, **kwargs: devices.unet_needs_upcast # pylint: disable=unnecessary-lambda-assignment