Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -146,6 +146,7 @@ def run_detailgen3d(
|
|
| 146 |
mesh = trimesh.Trimesh(vertices.astype(np.float32), np.ascontiguousarray(faces))
|
| 147 |
return mesh
|
| 148 |
|
|
|
|
| 149 |
@torch.no_grad()
|
| 150 |
@torch.autocast(device_type=DEVICE)
|
| 151 |
def run_refinement(
|
|
|
|
| 146 |
mesh = trimesh.Trimesh(vertices.astype(np.float32), np.ascontiguousarray(faces))
|
| 147 |
return mesh
|
| 148 |
|
| 149 |
+
@spaces.GPU()
|
| 150 |
@torch.no_grad()
|
| 151 |
@torch.autocast(device_type=DEVICE)
|
| 152 |
def run_refinement(
|