File size: 548 Bytes
aa89e50 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | {
"_class_name": "WanTransformer3DModel",
"_diffusers_version": "0.36.0",
"_name_or_path": "/media/damoxing/ckp/robbyant/lingbot-va-posttrain-robotwin-for-train/transformer",
"action_dim": 30,
"attention_head_dim": 128,
"attn_mode": "flex",
"cross_attn_norm": true,
"eps": 1e-06,
"ffn_dim": 14336,
"freq_dim": 256,
"in_channels": 48,
"num_attention_heads": 24,
"num_layers": 30,
"out_channels": 48,
"patch_size": [
1,
2,
2
],
"pos_embed_seq_len": null,
"rope_max_seq_len": 1024,
"text_dim": 4096
}
|