| { |
| "target": "partgen.models.conditioner.condioner_release.Conditioner", |
| "params": { |
| "use_geo": true, |
| "use_obj": true, |
| "use_seg_feat": true, |
| "geo_cfg": { |
| "target": "partgen.models.conditioner.part_encoders.PartEncoder", |
| "output_dim": 1024, |
| "params": { |
| "use_local": true, |
| "local_feat_type": "latents_shape", |
| "num_tokens_cond": 4096, |
| "local_geo_cfg": { |
| "target": "partgen.models.autoencoders.VolumeDecoderShapeVAE", |
| "params": { |
| "num_latents": 4096, |
| "embed_dim": 64, |
| "num_freqs": 8, |
| "include_pi": false, |
| "heads": 16, |
| "width": 1024, |
| "num_encoder_layers": 8, |
| "num_decoder_layers": 16, |
| "qkv_bias": false, |
| "qk_norm": true, |
| "scale_factor": null, |
| "geo_decoder_mlp_expand_ratio": 4, |
| "geo_decoder_downsample_ratio": 1, |
| "geo_decoder_ln_post": true, |
| "point_feats": 4, |
| "pc_size": 81920, |
| "pc_sharpedge_size": 0 |
| } |
| } |
| } |
| }, |
| "obj_encoder_cfg": { |
| "target": "partgen.models.autoencoders.VolumeDecoderShapeVAE", |
| "output_dim": 1024, |
| "params": { |
| "num_latents": 4096, |
| "embed_dim": 64, |
| "num_freqs": 8, |
| "include_pi": false, |
| "heads": 16, |
| "width": 1024, |
| "num_encoder_layers": 8, |
| "num_decoder_layers": 16, |
| "qkv_bias": false, |
| "qk_norm": true, |
| "scale_factor": 1.0039506158752403, |
| "geo_decoder_mlp_expand_ratio": 4, |
| "geo_decoder_downsample_ratio": 1, |
| "geo_decoder_ln_post": true, |
| "point_feats": 4, |
| "pc_size": 81920, |
| "pc_sharpedge_size": 0 |
| } |
| }, |
| "seg_feat_cfg": { |
| "target": "partgen.models.conditioner.sonata_extractor.SonataFeatureExtractor" |
| } |
| } |
| } |
