update

asset3d_gen/data/differentiable_render.py

@@ -400,13 +400,6 @@ def parse_args():
         default=1.0,
         help="Light factor for mesh PBR rendering (default: 2.)",
     )
-    parser.add_argument(
-        "--device",
-        type=str,
-        choices=["cpu", "cuda"],
-        default="cuda",
-        help="Device to run on (default: 'cuda')",
-    )
     parser.add_argument(
         "--with_mtl",
         action="store_true",
@@ -463,7 +456,7 @@ def entrypoint(**kwargs) -> None:
         distance=args.distance,
         resolution_hw=args.resolution_hw,
         fov=math.radians(args.fov),
-        device=args.device,
+        device="cuda",
     )
 
     render_items = [

asset3d_gen/utils/process_media.py

@@ -6,7 +6,7 @@ import subprocess
 from glob import glob
 from io import BytesIO
 from typing import Union
-
+import spaces
 import cv2
 import imageio
 import numpy as np
@@ -26,6 +26,7 @@ __all__ = [
 ]
 
 
+@spaces.GPU
 def render_asset3d(
     mesh_path: str,
     output_root: str,
@@ -38,7 +39,6 @@ def render_asset3d(
     gen_color_mp4: bool = False,
     gen_viewnormal_mp4: bool = False,
     gen_glonormal_mp4: bool = False,
-    device: str = "cpu",
 ) -> list[str]:
     command = [
         "python3",
@@ -58,8 +58,6 @@ def render_asset3d(
         "--pbr_light_factor",
         str(pbr_light_factor),
         "--with_mtl",
-        "--device",
-        device,
     ]
     if gen_color_mp4:
         command.append("--gen_color_mp4")