| { | |
| "model_type": "flux", | |
| "framework": "jax", | |
| "library": "diffuse", | |
| "architecture": "FluxTransformer2D", | |
| "version": "1.0-dev", | |
| "components": { | |
| "transformer": { | |
| "hidden_dim": 3072, | |
| "num_heads": 24, | |
| "num_double_layers": 19, | |
| "num_single_layers": 38, | |
| "in_channels": 64 | |
| }, | |
| "text_encoder": "clip", | |
| "text_encoder_2": "t5", | |
| "vae": "autoencoder_kl" | |
| }, | |
| "precision": "bfloat16", | |
| "task": "text-to-image" | |
| } | |