update modeling.py (#5)
Browse files- update modeling.py (a4c9a3ccbca6d37fcdfab04b9c07512b2366968d)
Co-authored-by: Junlin Han <[email protected]>
- 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=
|
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]
|