Spaces:

Zhengyi
/

CRM

Running on Zero

Zhengyi commited on Mar 11

Commit

d56e314

•

1 Parent(s): 78b51a8

float32

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -37,7 +37,7 @@ def generate3d(model, rgb, ccm, device):
     triplane = torch.cat([color,xyz],dim=1).to(device)
     # 3D visualize
     model.eval()
-    glctx = dr.RasterizeCudaContext()
     if model.denoising == True:
         tnew = 20

     triplane = torch.cat([color,xyz],dim=1).to(device)
     # 3D visualize
     model.eval()
+    glctx = dr.RasterizeGLContext()#dr.RasterizeCudaContext()
     if model.denoising == True:
         tnew = 20