| { | |
| "bands": 8, | |
| "cond_bands": 8, | |
| "cond_dim": 1152, | |
| "cond_heads": 12, | |
| "cond_layers": 4, | |
| "depth": 24, | |
| "dim": 1152, | |
| "heads": 16, | |
| "in_channels": 1, | |
| "latent_scale": 1.0, | |
| "latent_shift": 0.0, | |
| "latent_size": 64, | |
| "mlp_ratio": 4.0, | |
| "patch_size": 4, | |
| "token_dim": 768 | |
| } |