| { | |
| "name": "UniLatEncoder", | |
| "args": { | |
| "resolution": 64, | |
| "in_channels": 1024, | |
| "model_channels": 768, | |
| "latent_channels": 8, | |
| "num_blocks": 12, | |
| "num_heads": 12, | |
| "mlp_ratio": 4, | |
| "attn_mode": "swin", | |
| "window_size": 8, | |
| "use_fp16": true, | |
| "feats_latent_channels": 32, | |
| "num_res_blocks": 4, | |
| "pe_voxel_dim": 0 | |
| } | |
| } |