| { |
| "geometry_checkpoint_dir": "weights/vomp_1_1/geometry_transformer.pt", |
| "matvae_checkpoint_dir": "weights/vomp_1_1/matvae.safetensors", |
| "normalization_params_path": "weights/vomp_1_1/normalization_params.json", |
| "matvae_checkpoint": "weights/vomp_1_1/matvae.safetensors", |
| "trellis_weights_path": "weights/TRELLIS-image-large", |
| "models": { |
| "geometry_encoder": { |
| "name": "geometry_encoder", |
| "args": { |
| "resolution": 64, |
| "in_channels": 1024, |
| "model_channels": 768, |
| "latent_channels": 2, |
| "num_blocks": 12, |
| "num_heads": 12, |
| "mlp_ratio": 4, |
| "attn_mode": "swin", |
| "window_size": 8, |
| "use_fp16": true |
| } |
| }, |
| "matvae": { |
| "name": "StandardVAE", |
| "args": { |
| "width": 256, |
| "depth": 3, |
| "z_dim": 2, |
| "p_drop": 0.05, |
| "use_learned_variances": false, |
| "use_additional_losses": true, |
| "use_flow": false |
| } |
| } |
| }, |
| "dataset": { |
| "name": "SparseVoxelMaterials", |
| "normalization_type": "log_minmax", |
| "args": { |
| "roots": "datasets/simready", |
| "image_size": 512, |
| "model": "dinov2_vitl14_reg", |
| "resolution": 64, |
| "min_aesthetic_score": 0.0, |
| "max_num_voxels": 32768, |
| "compute_material_stats": false, |
| "archive_format": "mapped", |
| "random_grid_rotation": false, |
| "segment_balanced_sampling": true |
| } |
| } |
| } |
|
|
