transformer/config.json

{
    "_class_name": "WanTransformer3DModel",
    "added_kv_proj_dim": null,
    "attention_head_dim": 128,
    "cross_attn_norm": true,
    "eps": 1e-06,
    "ffn_dim": 13824,
    "freq_dim": 256,
    "image_dim": null,
    "in_channels": 16,
    "num_attention_heads": 40,
    "num_layers": 40,
    "out_channels": 16,
    "patch_size": [
        1,
        2,
        2
    ],
    "qk_norm": "rms_norm_across_heads",
    "rope_max_seq_len": 1024,
    "text_dim": 4096
}