File size: 529 Bytes
2fe3da0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
model_config:
target: src.models.lrm_mesh.PRM
params:
encoder_feat_dim: 768
encoder_freeze: false
encoder_model_name: facebook/dino-vitb16
transformer_dim: 1024
transformer_layers: 16
transformer_heads: 16
triplane_low_res: 32
triplane_high_res: 64
triplane_dim: 80
rendering_samples_per_ray: 128
grid_res: 128
grid_scale: 2.1
infer_config:
unet_path: ckpts/diffusion_pytorch_model.bin
model_path: ckpts/final_ckpt.ckpt
texture_resolution: 2048
render_resolution: 512 |