diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-12-10 05:45:41 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-12-10 05:45:41 +0000 |
| commit | 2641d1b83bca6961ab1367fd9b67ebd8a94af35b (patch) | |
| tree | c168105741bd7a3b958f080b57bd3088fc74f04e /modules/sd_hijack.py | |
| parent | 4d5fe3bfc01568d6cd6d66c7f3de5c714776cf7b (diff) | |
| parent | 3a724e91a23a715987e3034f31d8c8f5d416bd3a (diff) | |
| download | stable-diffusion-webui-gfx803-2641d1b83bca6961ab1367fd9b67ebd8a94af35b.tar.gz stable-diffusion-webui-gfx803-2641d1b83bca6961ab1367fd9b67ebd8a94af35b.tar.bz2 stable-diffusion-webui-gfx803-2641d1b83bca6961ab1367fd9b67ebd8a94af35b.zip | |
Merge pull request #4978 from aliencaocao/support_any_resolution
Patch UNet Forward to support resolutions that are not multiples of 64
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.")
|