Skip to content

S
Stable Diffusion Webui
Commit cc0258ae authored by C43H66N12O12S2's avatar C43H66N12O12S2 Committed by AUTOMATIC1111
Browse files
Options

check for ampere without destroying the optimizations. again.

parent 017b6b87
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 4 deletions
modules/sd_hijack.py
View file @ cc0258ae
...@@ -22,8 +22,7 @@ def apply_optimizations(): ...@@ -22,8 +22,7 @@ def apply_optimizations():
undo_optimizations() undo_optimizations()
ldm.modules.diffusionmodules.model.nonlinearity = silu ldm.modules.diffusionmodules.model.nonlinearity = silu
if cmd_opts.xformers and shared.xformers_available and torch.version.cuda: if cmd_opts.xformers and shared.xformers_available and torch.version.cuda and torch.cuda.get_device_capability(shared.device) == (8, 6):
if torch.cuda.get_device_capability(shared.device) == (8, 6):
ldm.modules.attention.CrossAttention.forward = sd_hijack_optimizations.xformers_attention_forward ldm.modules.attention.CrossAttention.forward = sd_hijack_optimizations.xformers_attention_forward
ldm.modules.diffusionmodules.model.AttnBlock.forward = sd_hijack_optimizations.xformers_attnblock_forward ldm.modules.diffusionmodules.model.AttnBlock.forward = sd_hijack_optimizations.xformers_attnblock_forward
elif cmd_opts.opt_split_attention_v1: elif cmd_opts.opt_split_attention_v1:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment