LTT commited on
Commit
54e6d5f
·
verified ·
1 Parent(s): 518e8c7

Update models/ISOMER/scripts/project_mesh.py

Browse files
models/ISOMER/scripts/project_mesh.py CHANGED
@@ -94,6 +94,7 @@ class Pix2FacesRenderer:
94
  @spaces.GPU
95
  def render_pix2faces_nvdiff(self, meshes: Meshes, cameras: CamerasBase, H=512, W=512):
96
  self.device = "cuda"
 
97
  meshes = meshes.to(self.device)
98
  cameras = cameras.to(self.device)
99
  vertices = self.transform_vertices(meshes, cameras)
 
94
  @spaces.GPU
95
  def render_pix2faces_nvdiff(self, meshes: Meshes, cameras: CamerasBase, H=512, W=512):
96
  self.device = "cuda"
97
+ self._glctx = dr.RasterizeCudaContext(device=self.device)
98
  meshes = meshes.to(self.device)
99
  cameras = cameras.to(self.device)
100
  vertices = self.transform_vertices(meshes, cameras)