diff options
Diffstat (limited to 'modules/sd_hijack.py')
-rw-r--r-- | modules/sd_hijack.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py index 95a17093..92874a79 100644 --- a/modules/sd_hijack.py +++ b/modules/sd_hijack.py @@ -39,6 +39,7 @@ def apply_optimizations(): undo_optimizations()
ldm.modules.diffusionmodules.model.nonlinearity = silu
+ ldm.modules.diffusionmodules.openaimodel.UNetModel.forward = sd_hijack_optimizations.patched_unet_forward
if cmd_opts.force_enable_xformers or (cmd_opts.xformers and shared.xformers_available and torch.version.cuda and (6, 0) <= torch.cuda.get_device_capability(shared.device) <= (9, 0)):
print("Applying xformers cross attention optimization.")
|