From 89aae3e04fe6aa11c3da76b9b48eed16be310b14 Mon Sep 17 00:00:00 2001
From: Kohya S <ykumeykume@gmail.com>
Date: Mon, 31 Jul 2023 21:48:19 +0900
Subject: [PATCH] fix vae crashes in large reso

---
 sdxl_gen_img.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/sdxl_gen_img.py b/sdxl_gen_img.py
index 12ab46f7..d3d7f074 100644
--- 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()