From 686791ef7018b339e5f1c0958cf2fc352542bc1b Mon Sep 17 00:00:00 2001
From: sylym <qwejkl2222ccyy@126.com>
Date: Fri, 24 Mar 2023 22:14:49 +0800
Subject: [PATCH] first modification

---
 sd.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/sd.py b/sd.py
index 4b7e64f..332f4d6 100644
--- a/sd.py
+++ b/sd.py
@@ -104,6 +104,9 @@ def load_checkpoint_guess_config(ckpt_path, output_vae=True, output_clip=True, e
     model = load_model_weights(model, sd, verbose=False, load_state_dict_to=load_state_dict_to)
     model.model.diffusion_model = convert_unet_checkpoint(sd, OmegaConf.create({"model": model_config}))
 
+    if model_management.xformers_enabled():
+        model.model.diffusion_model.enable_xformers_memory_efficient_attention()
+
     if fp16:
        model = model.half()