{
  "kind": "diffusion",
  "version": "0.2",
  "diffusion": {
    "model_id": "black-forest-labs/FLUX.2-klein-4B",
    "type": "flux2",
    "prediction_type": "flow_matching",
    "encoder_scale_factor": 1.0,
    "decoder_scale_factor": 1.0,
    "decoder_shift_factor": 0.0,
    "batch_norm_eps": 0.0001,
    "guidance_embeds": false,
    "image_size": 1024,
    "default_guidance_scale": 1.0,
    "default_steps": 4,
    "rope_axes_dims": [32, 32, 32, 32],
    "rope_theta": 2000
  },
  "assets": {
    "text_encoder": "TextEncoder.aimodel",
    "transformer": "Transformer.aimodel",
    "vae_decoder": "VAEDecoder.aimodel",
    "vae_encoder": "VAEEncoder.aimodel"
  }
}