Update models/ISOMER/scripts/project_mesh.py

models/ISOMER/scripts/project_mesh.py

@@ -94,6 +94,7 @@ class Pix2FacesRenderer:
     @spaces.GPU
     def render_pix2faces_nvdiff(self, meshes: Meshes, cameras: CamerasBase, H=512, W=512):
         self.device = "cuda"
+        self._glctx = dr.RasterizeCudaContext(device=self.device)
         meshes = meshes.to(self.device)
         cameras = cameras.to(self.device)
         vertices = self.transform_vertices(meshes, cameras)