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)
|