Update models/ISOMER/mesh_reconstruction/render.py

models/ISOMER/mesh_reconstruction/render.py

@@ -4,6 +4,13 @@ import torch
 from typing import Tuple
 import spaces
 
+glctx = None
+@spaces.GPU
+def init_render():
+    global glctx
+    glctx = dr.RasterizeCudaContext(device="cuda")
+init_render()
+
 def _warmup(glctx, device=None):
     device = 'cuda' if device is None else device
     #windows workaround for https://github.com/NVlabs/nvdiffrast/issues/59
@@ -16,12 +23,7 @@ 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
-def init_render():
-    global glctx
-    glctx = dr.RasterizeCudaContext(device="cuda")
-init_render()
+
 class NormalsRenderer:
     
     _glctx:dr.RasterizeCudaContext = None