{
  "_class_name": "MageFlow",
  "apply_text_rotary_emb": false,
  "axes_dim": [
    16,
    56,
    56
  ],
  "checkpoint": false,
  "context_in_dim": 2560,
  "depth": 12,
  "depth_single_blocks": 0,
  "double_block_type": "double_stream",
  "guidance_embed": false,
  "hidden_size": 3072,
  "in_channels": 128,
  "max_sequence_length": 2048,
  "mlp_ratio": 4.0,
  "num_heads": 24,
  "out_channels": 128,
  "packing": true,
  "param_dtype": "bfloat16",
  "patch_size": 1,
  "qkv_bias": true,
  "quantization_config": {
    "backend": "native_sm120",
    "compute_dtype": "bfloat16",
    "format": "e2m1_block16_ue4m3",
    "quant_algo": "NVFP4",
    "quant_method": "mage_flow_nvfp4",
    "target_count": 48,
    "targets": [
      "transformer_blocks.0.img_mlp.net.0.proj",
      "transformer_blocks.0.img_mlp.net.2",
      "transformer_blocks.0.txt_mlp.net.0.proj",
      "transformer_blocks.0.txt_mlp.net.2",
      "transformer_blocks.1.img_mlp.net.0.proj",
      "transformer_blocks.1.img_mlp.net.2",
      "transformer_blocks.1.txt_mlp.net.0.proj",
      "transformer_blocks.1.txt_mlp.net.2",
      "transformer_blocks.2.img_mlp.net.0.proj",
      "transformer_blocks.2.img_mlp.net.2",
      "transformer_blocks.2.txt_mlp.net.0.proj",
      "transformer_blocks.2.txt_mlp.net.2",
      "transformer_blocks.3.img_mlp.net.0.proj",
      "transformer_blocks.3.img_mlp.net.2",
      "transformer_blocks.3.txt_mlp.net.0.proj",
      "transformer_blocks.3.txt_mlp.net.2",
      "transformer_blocks.4.img_mlp.net.0.proj",
      "transformer_blocks.4.img_mlp.net.2",
      "transformer_blocks.4.txt_mlp.net.0.proj",
      "transformer_blocks.4.txt_mlp.net.2",
      "transformer_blocks.5.img_mlp.net.0.proj",
      "transformer_blocks.5.img_mlp.net.2",
      "transformer_blocks.5.txt_mlp.net.0.proj",
      "transformer_blocks.5.txt_mlp.net.2",
      "transformer_blocks.6.img_mlp.net.0.proj",
      "transformer_blocks.6.img_mlp.net.2",
      "transformer_blocks.6.txt_mlp.net.0.proj",
      "transformer_blocks.6.txt_mlp.net.2",
      "transformer_blocks.7.img_mlp.net.0.proj",
      "transformer_blocks.7.img_mlp.net.2",
      "transformer_blocks.7.txt_mlp.net.0.proj",
      "transformer_blocks.7.txt_mlp.net.2",
      "transformer_blocks.8.img_mlp.net.0.proj",
      "transformer_blocks.8.img_mlp.net.2",
      "transformer_blocks.8.txt_mlp.net.0.proj",
      "transformer_blocks.8.txt_mlp.net.2",
      "transformer_blocks.9.img_mlp.net.0.proj",
      "transformer_blocks.9.img_mlp.net.2",
      "transformer_blocks.9.txt_mlp.net.0.proj",
      "transformer_blocks.9.txt_mlp.net.2",
      "transformer_blocks.10.img_mlp.net.0.proj",
      "transformer_blocks.10.img_mlp.net.2",
      "transformer_blocks.10.txt_mlp.net.0.proj",
      "transformer_blocks.10.txt_mlp.net.2",
      "transformer_blocks.11.img_mlp.net.0.proj",
      "transformer_blocks.11.img_mlp.net.2",
      "transformer_blocks.11.txt_mlp.net.0.proj",
      "transformer_blocks.11.txt_mlp.net.2"
    ]
  },
  "rope_type": "msrope",
  "schedule_mode": "z-image",
  "static_shift": 6.0,
  "theta": 10000,
  "time_type": "qwen_proj",
  "txt_max_length": 2048,
  "use_time_shift": false,
  "vec_in_dim": 0,
  "vec_type": null
}