Merge pull request #4978 from aliencaocao/support_any_resolution

Patch UNet Forward to support resolutions that are not multiples of 64
This commit is contained in:
AUTOMATIC1111
2022-12-10 08:45:41 +03:00
committed by GitHub
3 changed files with 44 additions and 12 deletions

View File

@@ -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.")