Reverb commited on
Commit
68a4e06
·
1 Parent(s): 5dad3a0

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,