Update models/ISOMER/mesh_reconstruction/render.py

models/ISOMER/mesh_reconstruction/render.py

@@ -16,9 +16,12 @@ def _warmup(glctx, device=None):
     dr.rasterize(glctx, pos, tri, resolution=[256, 256])
 
 # glctx = dr.RasterizeGLContext(output_db=False, device="cuda")
+glctx = None
 @spaces.GPU
-glctx = dr.RasterizeCudaContext(device="cuda")
-
+def init_render():
+    global glctx
+    glctx = dr.RasterizeCudaContext(device="cuda")
+init_render()
 class NormalsRenderer:
     
     _glctx:dr.RasterizeCudaContext = None