Spaces:

LTT
/

Kiss3DGen

Runtime error

LTT commited on Dec 17, 2024

Commit

cdb6265

verified ·

1 Parent(s): 7504eb0

Update models/ISOMER/mesh_reconstruction/render.py

Files changed (1) hide show

models/ISOMER/mesh_reconstruction/render.py CHANGED Viewed

@@ -13,7 +13,9 @@ init_render()
 @spaces.GPU
 def _warmup(glctx, device=None):
-    global glctx
     device = 'cuda' if device is None else device
     #windows workaround for https://github.com/NVlabs/nvdiffrast/issues/59
     def tensor(*args, **kwargs):

 @spaces.GPU
 def _warmup(glctx, device=None):
+    if glctx==None:
+        print("==> glctx in none")
+        glctx = dr.RasterizeCudaContext(device="cuda")
     device = 'cuda' if device is None else device
     #windows workaround for https://github.com/NVlabs/nvdiffrast/issues/59
     def tensor(*args, **kwargs):