| { | |
| "model_type": "dense_dit", | |
| "model_class": "pixal3d.models.transformers.dense_dit.DenseDiTDenoiser", | |
| "resolution": 16, | |
| "in_channels": 8, | |
| "out_channels": 8, | |
| "model_channels": 1024, | |
| "cond_channels": 1024, | |
| "num_blocks": 24, | |
| "num_heads": 16, | |
| "mlp_ratio": 4, | |
| "patch_size": 1, | |
| "pe_mode": "ape", | |
| "qk_rms_norm": true, | |
| "use_checkpoint": true, | |
| "use_fp16": false, | |
| "latent_shape": [ | |
| 8, | |
| 16, | |
| 16, | |
| 16 | |
| ], | |
| "image_attn_mode": "proj" | |
| } |