1 files changed, 1 insertions, 0 deletions

diff --git a/infer.py b/infer.py
index eabeb5e..75c8621 100644
--- a/infer.py
+++ b/infer.py
@@ -219,6 +219,7 @@ def create_pipeline(model, ti_embeddings_dir, dtype):
         scheduler=scheduler,
     )
     pipeline.enable_xformers_memory_efficient_attention()
+    pipeline.enable_vae_slicing()
     pipeline.to("cuda")
 
     print("Pipeline loaded.")