PRM / configs /PRM_inference.yaml
JiantaoLin
new
2fe3da0
raw
history blame
529 Bytes
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