| { | |
| "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" | |
| } | |
| } | |
| } |