dit-config.yaml

depth: 48
patch_size: 2
num_heads: 24
hidden_size_x: 3072
hidden_size_y: 1536
mlp_ratio_x: 4.0
mlp_ratio_y: 4.0
in_channels: 12
qk_norm: true
qkv_bias: false
out_bias: true
patch_embed_bias: true
timestep_mlp_bias: true
timestep_scale: 1000.0
t5_feat_dim: 4096
t5_token_length: 256
rope_theta: 10000.0