VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix vae crashes in large reso

Browse Source
This commit is contained in:
Kohya S
2023-07-31 21:48:19 +09:00
parent 0636399c8c
commit 89aae3e04f
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
sdxl_gen_img.py
View File

@@ -93,7 +93,8 @@ def replace_vae_modules(vae: diffusers.models.AutoencoderKL, mem_eff_attn, xform
if mem_eff_attn: if mem_eff_attn:
replace_vae_attn_to_memory_efficient() replace_vae_attn_to_memory_efficient()
elif xformers: elif xformers:
replace_vae_attn_to_xformers() # replace_vae_attn_to_xformers() # 解像度によってxformersがエラーを出す？
vae.set_use_memory_efficient_attention_xformers(True) # とりあえずこっちを使う
elif sdpa: elif sdpa:
replace_vae_attn_to_sdpa() replace_vae_attn_to_sdpa()