update

infer_api.py

@@ -767,9 +767,11 @@ infer_multiview_device = torch.device("cuda:0" if torch.cuda.is_available() else
 infer_multiview_pipeline = StableUnCLIPImg2ImgPipeline.from_pretrained(
     infer_multiview_cfg.pretrained_path,
     torch_dtype=torch.float16,)
-infer_multiview_pipeline.unet.enable_xformers_memory_efficient_attention()
 if torch.cuda.is_available():
     infer_multiview_pipeline.to(device)
+    infer_multiview_pipeline.unet.enable_xformers_memory_efficient_attention()
+    print(f"Era3D Using device!!!!!!!!!!!!: {device}", file=sys.stderr)
+
 infer_multiview_results = {}
 
 infer_multiview_image_transforms = [transforms.Resize(int(max(infer_multiview_cfg.height, infer_multiview_cfg.width))),