update

refine/func.py

@@ -134,7 +134,7 @@ def from_py3d_mesh(mesh):
 
 class Pix2FacesRenderer:
     def __init__(self, device="cuda"):
-        self._glctx = dr.RasterizeGLContext(device=device)
+        self._glctx = dr.RasterizeGLContext()
         self.device = device
         _warmup(self._glctx, device)
 

refine/render.py

@@ -14,7 +14,7 @@ def _warmup(glctx, device=None):
     tri = tensor([[0, 1, 2]], dtype=torch.int32)
     dr.rasterize(glctx, pos, tri, resolution=[256, 256])
 
-glctx = dr.RasterizeGLContext(device="cuda")
+glctx = dr.RasterizeGLContext()
 
 class NormalsRenderer: