CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

force

Browse Source
This commit is contained in:
Haoming
2025-12-03 10:22:00 +08:00
parent d975833292
commit ab5c12eb43
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/args.py
+2
View File
@@ -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")
backend/attention.py
+1 -1
View File
@@ -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():