Spaces:
mashroo
/
Runtime error

YoussefAnso commited on
Commit
a3626f7
·
1 Parent(s): db2fd1d

Update generate3d function to initialize nvdiffrast context using CUDA, enhancing rendering performance and compatibility with GPU devices.

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -21,7 +21,7 @@ def generate3d(model, rgb_image, xyz_image, device="cpu"):
21
  # Get rendering context required by xatlas and nvdiffrast
22
  try:
23
  import nvdiffrast.torch as dr
24
- ctx = dr.RasterizeGLContext()
25
  except Exception as e:
26
  raise RuntimeError("Failed to initialize nvdiffrast context: " + str(e))
27
 
 
21
  # Get rendering context required by xatlas and nvdiffrast
22
  try:
23
  import nvdiffrast.torch as dr
24
+ ctx = dr.RasterizeCudaContext(device=device)
25
  except Exception as e:
26
  raise RuntimeError("Failed to initialize nvdiffrast context: " + str(e))
27