Update models/ISOMER/scripts/project_mesh.py
Browse files
models/ISOMER/scripts/project_mesh.py
CHANGED
|
@@ -93,6 +93,7 @@ class Pix2FacesRenderer:
|
|
| 93 |
return vertices
|
| 94 |
@spaces.GPU
|
| 95 |
def render_pix2faces_nvdiff(self, meshes: Meshes, cameras: CamerasBase, H=512, W=512):
|
|
|
|
| 96 |
meshes = meshes.to(self.device)
|
| 97 |
cameras = cameras.to(self.device)
|
| 98 |
vertices = self.transform_vertices(meshes, cameras)
|
|
|
|
| 93 |
return vertices
|
| 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)
|