{
  "_class_name": "IMFTransformer2DModel",
  "_diffusers_version": "0.38.0",
  "aux_head_depth": 8,
  "depth": 12,
  "embedding_init_constant": 1.0,
  "eval_mode": true,
  "hidden_size": 768,
  "in_channels": 4,
  "mlp_ratio": 2.6666666666666665,
  "model_type": "iMF-B/2",
  "num_attention_heads": 12,
  "num_cfg_tokens": 4,
  "num_class_embeds": null,
  "num_class_tokens": 8,
  "num_classes": 1000,
  "num_interval_tokens": 2,
  "num_time_tokens": 4,
  "patch_size": 2,
  "sample_size": 32,
  "token_init_constant": 1.0,
  "weight_init_constant": 0.32
}