1 files changed, 1 insertions, 1 deletions

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.")