diff options
author | AUTOMATIC <16777216c@gmail.com> | 2022-10-08 17:02:18 +0300 |
---|---|---|
committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-08 17:02:18 +0300 |
commit | dc1117233ef8f9b25ff1ac40b158f20b70ba2fcb (patch) | |
tree | d7b54aa4cd8216984bc842323cc880baff3065c3 /modules/sd_hijack.py | |
parent | 7ff1170a2e11b6f00f587407326db0b9f8f51adf (diff) |
simplify xfrmers options: --xformers to enable and that's it
Diffstat (limited to 'modules/sd_hijack.py')
-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 5d93f7f6..91e98c16 100644 --- a/modules/sd_hijack.py +++ b/modules/sd_hijack.py @@ -22,7 +22,7 @@ def apply_optimizations(): undo_optimizations()
ldm.modules.diffusionmodules.model.nonlinearity = silu
- if not cmd_opts.disable_opt_xformers_attention and not (cmd_opts.opt_split_attention or torch.version.hip) and shared.xformers_available:
+ if cmd_opts.xformers and shared.xformers_available and not torch.version.hip:
ldm.modules.attention.CrossAttention.forward = sd_hijack_optimizations.xformers_attention_forward
ldm.modules.diffusionmodules.model.AttnBlock.forward = sd_hijack_optimizations.xformers_attnblock_forward
elif cmd_opts.opt_split_attention_v1:
|