update

infer_api.py

@@ -283,6 +283,8 @@ def save_image_numpy(ndarr):
 
 @spaces.GPU
 def run_multiview_infer(data, pipeline, cfg: TestConfig, num_levels=3):
+    infer_multiview_pipeline.unet.enable_xformers_memory_efficient_attention()
+    
     if cfg.seed is None:
         generator = None
     else:
@@ -769,7 +771,6 @@ infer_multiview_pipeline = StableUnCLIPImg2ImgPipeline.from_pretrained(
 if torch.cuda.is_available():
     infer_multiview_pipeline.to(infer_multiview_device)
     print(f"Era3D Using device!!!!!!!!!!!!: {infer_multiview_device}", file=sys.stderr)
-    infer_multiview_pipeline.unet.enable_xformers_memory_efficient_attention()
 
 infer_multiview_results = {}