fix exception related to the pix2pix (9c1c0da0) · Commits · github_fork / Stable Diffusion Webui

modules/sd_hijack.py

+4 −0

Original line number	Diff line number	Diff line
		@@ -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: