1 files changed, 2 insertions, 2 deletions

diff --git a/infer.py b/infer.py
index 13219f8..80bd208 100644
--- a/infer.py
+++ b/infer.py
@@ -245,8 +245,8 @@ def create_pipeline(model, dtype):
         tokenizer=tokenizer,
         scheduler=scheduler,
     )
-    # pipeline.enable_xformers_memory_efficient_attention()
-    pipeline.unet = torch.compile(pipeline.unet)
+    pipeline.enable_xformers_memory_efficient_attention()
+    # pipeline.unet = torch.compile(pipeline.unet)
     pipeline.enable_vae_slicing()
     pipeline.to("cuda")