{
  "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"
}