| { | |
| "architecture": "beatrix_flow_flux_style", | |
| "image_size": 256, | |
| "latent_channels": 16, | |
| "latent_size": 32, | |
| "text_raw_dim": 768, | |
| "hidden_size": 256, | |
| "tower_dim": 256, | |
| "tower_depth": 2, | |
| "num_heads": 8, | |
| "geometric_types": [ | |
| "cantor", | |
| "beatrix" | |
| ], | |
| "conv_types": [ | |
| "wide_resnet", | |
| "squeeze_excite" | |
| ], | |
| "manifold_dim": 4096, | |
| "num_flow_steps": 50, | |
| "num_latent_tokens": 1024 | |
| } |