From 036e33fbde6bad7c48bb6f6b3d695b7908535c64 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Thu, 3 Nov 2022 17:56:08 +0100
Subject: Update

---
 infer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'infer.py')

diff --git a/infer.py b/infer.py
index ac05955..9bc9efe 100644
--- a/infer.py
+++ b/infer.py
@@ -234,7 +234,7 @@ def create_pipeline(model, scheduler, ti_embeddings_dir, dtype):
         tokenizer=tokenizer,
         scheduler=scheduler,
     )
-    pipeline.aesthetic_gradient_iters = 20
+    pipeline.enable_xformers_memory_efficient_attention()
     pipeline.to("cuda")
 
     print("Pipeline loaded.")
-- 
cgit v1.2.3-54-g00ecf