{
  "_class_name": "HiDreamImageTransformer2DModel",
  "_diffusers_version": "0.34.0.dev0",
  "attention_head_dim": 8,
  "axes_dims_rope": [
    4,
    2,
    2
  ],
  "caption_channels": [
    32,
    16
  ],
  "force_inference_output": false,
  "in_channels": 4,
  "llama_layers": [
    0,
    1
  ],
  "max_resolution": [
    32,
    32
  ],
  "num_activated_experts": 2,
  "num_attention_heads": 4,
  "num_layers": 1,
  "num_routed_experts": 4,
  "num_single_layers": 1,
  "out_channels": 4,
  "patch_size": 2,
  "text_emb_dim": 64
}