From ab5c12eb43b2dab40fbb170bd3b33d2a3fbdc413 Mon Sep 17 00:00:00 2001 From: Haoming Date: Wed, 3 Dec 2025 10:22:00 +0800 Subject: [PATCH] force --- backend/args.py | 2 ++ backend/attention.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/backend/args.py b/backend/args.py index 06b8d17b..f09dd641 100644 --- a/backend/args.py +++ b/backend/args.py @@ -66,6 +66,8 @@ parser.add_argument("--disable-xformers", action="store_true") parser.add_argument("--disable-sage", action="store_true") parser.add_argument("--disable-flash", action="store_true") +parser.add_argument("--force-xformers-vae", action="store_true") + parser.add_argument("--directml", type=int, nargs="?", metavar="DIRECTML_DEVICE", const=-1) parser.add_argument("--disable-ipex-hijack", action="store_true") diff --git a/backend/attention.py b/backend/attention.py index b9f5e46b..ce9e82f2 100644 --- a/backend/attention.py +++ b/backend/attention.py @@ -561,7 +561,7 @@ def pytorch_attention_single_head_spatial(q, k, v): return out -if memory_management.xformers_enabled_vae(): +if memory_management.xformers_enabled_vae() or args.force_xformers_vae: print("Using xformers Attention for VAE") attention_function_single_head_spatial = xformers_attention_single_head_spatial elif memory_management.pytorch_attention_enabled():