diff options
| author | brkirch <brkirch@users.noreply.github.com> | 2023-01-06 06:33:15 +0000 |
|---|---|---|
| committer | brkirch <brkirch@users.noreply.github.com> | 2023-01-06 06:33:15 +0000 |
| commit | 5deb2a19ccea57a50252e8fcb07b4d17c6599def (patch) | |
| tree | 7c4e3a0a840478001dcb53361d7378d9a64220b8 | |
| parent | b95a4c0ce5ab9c414e0494193bfff665f45e9e65 (diff) | |
| download | stable-diffusion-webui-gfx803-5deb2a19ccea57a50252e8fcb07b4d17c6599def.tar.gz stable-diffusion-webui-gfx803-5deb2a19ccea57a50252e8fcb07b4d17c6599def.tar.bz2 stable-diffusion-webui-gfx803-5deb2a19ccea57a50252e8fcb07b4d17c6599def.zip | |
Allow Doggettx's cross attention opt without CUDA
| -rw-r--r-- | modules/sd_hijack.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py index ef25dadb..bd101e5b 100644 --- a/modules/sd_hijack.py +++ b/modules/sd_hijack.py @@ -50,7 +50,7 @@ def apply_optimizations(): print("Applying v1 cross attention optimization.")
ldm.modules.attention.CrossAttention.forward = sd_hijack_optimizations.split_cross_attention_forward_v1
optimization_method = 'V1'
- elif not cmd_opts.disable_opt_split_attention and (cmd_opts.opt_split_attention_invokeai or not torch.cuda.is_available()):
+ elif not cmd_opts.disable_opt_split_attention and (cmd_opts.opt_split_attention_invokeai or not cmd_opts.opt_split_attention and not torch.cuda.is_available()):
print("Applying cross attention optimization (InvokeAI).")
ldm.modules.attention.CrossAttention.forward = sd_hijack_optimizations.split_cross_attention_forward_invokeAI
optimization_method = 'InvokeAI'
|