Spaces:
Sleeping
Sleeping
FrozenBurning
commited on
Commit
•
59e2c71
1
Parent(s):
ef4bc82
force cuda dr
Browse files- inference.py +1 -1
inference.py
CHANGED
@@ -114,7 +114,7 @@ def extract_texmesh(args, model, output_path, device):
|
|
114 |
w0 = 1024
|
115 |
ssaa = 1
|
116 |
fp16 = True
|
117 |
-
glctx = dr.
|
118 |
v_np = vertices.astype(np.float32)
|
119 |
f_np = triangles.astype(np.int64)
|
120 |
v = torch.from_numpy(vertices).float().contiguous().to(device)
|
|
|
114 |
w0 = 1024
|
115 |
ssaa = 1
|
116 |
fp16 = True
|
117 |
+
glctx = dr.RasterizeCudaContext()
|
118 |
v_np = vertices.astype(np.float32)
|
119 |
f_np = triangles.astype(np.int64)
|
120 |
v = torch.from_numpy(vertices).float().contiguous().to(device)
|