Change to GPU

3D_Stage/lrm/models/exporters/mesh_exporter.py

@@ -78,7 +78,8 @@ class MeshExporter(Exporter):
 
     def configure(self, renderer: BaseRenderer) -> None:
         super().configure(renderer)
-        self.ctx = NVDiffRasterizerContext(self.cfg.context_type, self.device)
+        self.ctx = None
+        #self.ctx = NVDiffRasterizerContext(self.cfg.context_type, self.device)
         if self.cfg.fmt == "obj-mtl":
             lrm.warn(
                 f"fmt=obj-mtl is deprecated, please us fmt=obj and visual=uv instead."
@@ -101,7 +102,7 @@ class MeshExporter(Exporter):
     def get_texture_maps(
         self, scene_code: torch.Tensor, mesh: Mesh
     ) -> Dict[str, torch.Tensor]:
-        #self.ctx = NVDiffRasterizerContext(self.cfg.context_type, self.device)
+        self.ctx = NVDiffRasterizerContext(self.cfg.context_type, self.device)
         assert mesh.has_uv
         # clip space transform
         uv_clip = mesh.v_tex * 2.0 - 1.0

app.py

@@ -320,7 +320,7 @@ class Inference3D_API:
 
     @spaces.GPU(duration=120)
     def process_images(self, img_input0, img_input1, img_input2, img_input3, back_proj, smooth_iter):
-        #self.system = self.system.to("cuda")
+        self.system = self.system.to("cuda")
         meta = json.load(open("./3D_Stage/material/meta.json"))
         c2w_cond = [np.array(loc["transform_matrix"]) for loc in meta["locations"]]
         c2w_cond = torch.from_numpy(np.stack(c2w_cond, axis=0)).float()[None].to(self.device)