From c927eeb3516b8ecae201441a68956f32247a6d7c Mon Sep 17 00:00:00 2001 From: Volpeon Date: Fri, 17 Feb 2023 15:56:54 +0100 Subject: Back to xformers --- train_lora.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'train_lora.py') diff --git a/train_lora.py b/train_lora.py index 8a06ae8..330bcd6 100644 --- a/train_lora.py +++ b/train_lora.py @@ -421,8 +421,8 @@ def main(): args.pretrained_model_name_or_path) vae.enable_slicing() - # vae.set_use_memory_efficient_attention_xformers(True) - # unet.enable_xformers_memory_efficient_attention() + vae.set_use_memory_efficient_attention_xformers(True) + unet.enable_xformers_memory_efficient_attention() unet.to(accelerator.device, dtype=weight_dtype) text_encoder.to(accelerator.device, dtype=weight_dtype) -- cgit v1.2.3-54-g00ecf