Spaces:
Build error
Build error
- mesh_reconstruction/render.py +1 -1
- scripts/project_mesh.py +1 -1
mesh_reconstruction/render.py
CHANGED
|
@@ -29,7 +29,7 @@ class NormalsRenderer:
|
|
| 29 |
else:
|
| 30 |
self._mvp = mvp
|
| 31 |
self._image_size = image_size
|
| 32 |
-
self._glctx = dr.RasterizeCudaContext(
|
| 33 |
_warmup(self._glctx, device)
|
| 34 |
|
| 35 |
def render(self,
|
|
|
|
| 29 |
else:
|
| 30 |
self._mvp = mvp
|
| 31 |
self._image_size = image_size
|
| 32 |
+
self._glctx = dr.RasterizeCudaContext(device=device)
|
| 33 |
_warmup(self._glctx, device)
|
| 34 |
|
| 35 |
def render(self,
|
scripts/project_mesh.py
CHANGED
|
@@ -56,7 +56,7 @@ def _warmup(glctx, device=None):
|
|
| 56 |
|
| 57 |
class Pix2FacesRenderer:
|
| 58 |
def __init__(self, device="cuda"):
|
| 59 |
-
self._glctx = dr.RasterizeCudaContext(
|
| 60 |
self.device = device
|
| 61 |
_warmup(self._glctx, device)
|
| 62 |
|
|
|
|
| 56 |
|
| 57 |
class Pix2FacesRenderer:
|
| 58 |
def __init__(self, device="cuda"):
|
| 59 |
+
self._glctx = dr.RasterizeCudaContext(device=device)
|
| 60 |
self.device = device
|
| 61 |
_warmup(self._glctx, device)
|
| 62 |
|