michaelj commited on
Commit
3feaa5a
·
verified ·
1 Parent(s): 984dd78

change ident

Browse files
Files changed (1) hide show
  1. tsr/models/nerf_renderer.py +1 -1
tsr/models/nerf_renderer.py CHANGED
@@ -37,7 +37,7 @@ class TriplaneNeRFRenderer(BaseModule):
37
  chunk_size >= 0
38
  ), "chunk_size must be a non-negative integer (0 for no chunking)."
39
  self.chunk_size = chunk_size
40
- def make_step_grid(self,device, resolution: int, chunk_size: int = 32):
41
  coords = torch.linspace(-1.0, 1.0, resolution, device = device)
42
  x, y, z = torch.meshgrid(coords[0:chunk_size], coords, coords, indexing="ij")
43
  x = x.reshape(-1, 1)
 
37
  chunk_size >= 0
38
  ), "chunk_size must be a non-negative integer (0 for no chunking)."
39
  self.chunk_size = chunk_size
40
+ def make_step_grid(self,device, resolution: int, chunk_size: int = 32):
41
  coords = torch.linspace(-1.0, 1.0, resolution, device = device)
42
  x, y, z = torch.meshgrid(coords[0:chunk_size], coords, coords, indexing="ij")
43
  x = x.reshape(-1, 1)