update

asset3d_gen/data/differentiable_render.py

@@ -90,7 +90,6 @@ class ImageRender(object):
         render_items: list[RenderItems],
         camera_params: CameraSetting,
         recompute_vtx_normal: bool = True,
-        device: str = "cuda",
         with_mtl: bool = False,
         gen_color_gif: bool = False,
         gen_color_mp4: bool = False,
@@ -99,7 +98,6 @@ class ImageRender(object):
         no_index_file: bool = False,
         light_factor: float = 1.0,
     ) -> None:
-        camera_params.device = device
         camera = init_kal_camera(camera_params)
         self.camera = camera
 
@@ -125,7 +123,7 @@ class ImageRender(object):
         self.renderer = renderer
         self.recompute_vtx_normal = recompute_vtx_normal
         self.render_items = render_items
-        self.device = device
+        self.device = camera_params.device
         self.with_mtl = with_mtl
         self.gen_color_gif = gen_color_gif
         self.gen_color_mp4 = gen_color_mp4