'update'

app.py

@@ -150,7 +150,6 @@ def load_era3d_pipeline(cfg):
         cfg.pretrained_model_name_or_path,
         torch_dtype=weight_dtype
     )
-    pipeline.unet.enable_xformers_memory_efficient_attention()
     # sys.main_lock = threading.Lock()
     return pipeline
 
@@ -298,7 +297,7 @@ def run_demo():
     pipeline = load_era3d_pipeline(cfg)
     torch.set_grad_enabled(False)
     pipeline.to('cuda')
-
+    pipeline.unet.enable_xformers_memory_efficient_attention()
     predictor = sam_init()