lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-11 16:09:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update sd_hijack.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 05:18:53 -08:00
parent 54cea07af5
commit 5714f79301
1 changed files with 1 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
modules/sd_hijack.py
View File

@@ -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