fix vae crashes in large reso

2026-04-08 22:35:09 +00:00 · 2023-07-31 21:48:19 +09:00
parent 0636399c8c
commit 89aae3e04f
1 changed files with 2 additions and 1 deletions
--- a/sdxl_gen_img.py
+++ b/sdxl_gen_img.py
@@ -93,7 +93,8 @@ def replace_vae_modules(vae: diffusers.models.AutoencoderKL, mem_eff_attn, xform
    if mem_eff_attn:
        replace_vae_attn_to_memory_efficient()
    elif xformers:
-        replace_vae_attn_to_xformers()
+        # replace_vae_attn_to_xformers() # 解像度によってxformersがエラーを出す？
+        vae.set_use_memory_efficient_attention_xformers(True) # とりあえずこっちを使う
    elif sdpa:
        replace_vae_attn_to_sdpa()