Spaces:
Sleeping
Sleeping
Remove remaining @spaces.GPU decorators from bake_albedo and bake_ao
Browse files
src/stages/stage2_bake_albedo.py
CHANGED
|
@@ -46,7 +46,6 @@ def _dilate(img: np.ndarray, mask: np.ndarray, n: int = 8) -> np.ndarray:
|
|
| 46 |
return result.clip(0, 255).astype(np.uint8)
|
| 47 |
|
| 48 |
|
| 49 |
-
@spaces.GPU(duration=60)
|
| 50 |
def bake_albedo(
|
| 51 |
high_poly_path: Path,
|
| 52 |
unwrapped_glb_path: Path,
|
|
@@ -126,7 +125,6 @@ def bake_albedo(
|
|
| 126 |
return out_path, f"Albedo baked: {map_size}×{map_size}"
|
| 127 |
|
| 128 |
|
| 129 |
-
@spaces.GPU(duration=60)
|
| 130 |
def bake_material(
|
| 131 |
high_poly_path: Path,
|
| 132 |
unwrapped_glb_path: Path,
|
|
|
|
| 46 |
return result.clip(0, 255).astype(np.uint8)
|
| 47 |
|
| 48 |
|
|
|
|
| 49 |
def bake_albedo(
|
| 50 |
high_poly_path: Path,
|
| 51 |
unwrapped_glb_path: Path,
|
|
|
|
| 125 |
return out_path, f"Albedo baked: {map_size}×{map_size}"
|
| 126 |
|
| 127 |
|
|
|
|
| 128 |
def bake_material(
|
| 129 |
high_poly_path: Path,
|
| 130 |
unwrapped_glb_path: Path,
|
src/stages/stage2_bake_ao.py
CHANGED
|
@@ -59,7 +59,6 @@ def _hemisphere_rays(normal: np.ndarray, n_rays: int, rng: np.random.Generator):
|
|
| 59 |
return local_dirs @ np.stack([T, B, normal]).T # [n_rays, 3]
|
| 60 |
|
| 61 |
|
| 62 |
-
@spaces.GPU(duration=120)
|
| 63 |
def bake_ao(
|
| 64 |
mesh_path: Path,
|
| 65 |
unwrapped_glb_path: Path,
|
|
|
|
| 59 |
return local_dirs @ np.stack([T, B, normal]).T # [n_rays, 3]
|
| 60 |
|
| 61 |
|
|
|
|
| 62 |
def bake_ao(
|
| 63 |
mesh_path: Path,
|
| 64 |
unwrapped_glb_path: Path,
|