jadechoghari JunlinHan commited on
Commit
733a173
1 Parent(s): 7841a02

update modeling.py (#5)

Browse files

- update modeling.py (a4c9a3ccbca6d37fcdfab04b9c07512b2366968d)


Co-authored-by: Junlin Han <JunlinHan@users.noreply.huggingface.co>

Files changed (1) hide show
  1. modeling.py +1 -1
modeling.py CHANGED
@@ -67,7 +67,7 @@ class LRMGenerator(PreTrainedModel):
67
  triplane_dim=config.triplane_dim, samples_per_ray=config.rendering_samples_per_ray,
68
  )
69
 
70
- def forward(self, image, camera, export_mesh=False, mesh_size=256, render_size=256, export_video=False, fps=30):
71
 
72
  assert image.shape[0] == camera.shape[0], "Batch size mismatch"
73
  N = image.shape[0]
 
67
  triplane_dim=config.triplane_dim, samples_per_ray=config.rendering_samples_per_ray,
68
  )
69
 
70
+ def forward(self, image, camera, export_mesh=False, mesh_size=512, render_size=384, export_video=False, fps=30):
71
 
72
  assert image.shape[0] == camera.shape[0], "Batch size mismatch"
73
  N = image.shape[0]