gitea-mirror/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-12-22 23:26:48 -08:00
Code Issues Packages Projects Releases Wiki Activity

fix exception related to the pix2pix

Browse Source
This commit is contained in:
AUTOMATIC1111
2023-11-06 11:17:36 +03:00
parent 656437e0a5
commit 9c1c0da026
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules/sd_hijack.py
View File

@@ -256,8 +256,12 @@ class StableDiffusionModelHijack:
self.layers = flatten(m) self.layers = flatten(m)
import modules.models.diffusion.ddpm_edit
if isinstance(m, ldm.models.diffusion.ddpm.LatentDiffusion): if isinstance(m, ldm.models.diffusion.ddpm.LatentDiffusion):
sd_unet.original_forward = ldm_original_forward 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): elif isinstance(m, sgm.models.diffusion.DiffusionEngine):
sd_unet.original_forward = sgm_original_forward sd_unet.original_forward = sgm_original_forward
else: else: