flowpro-round1-5000 / config.json
myendless's picture
Upload folder using huggingface_hub
912a330 verified
Raw
History Blame Contribute Delete
724 Bytes
{
"_class_name": "WanTransformer3DModel",
"_diffusers_version": "0.36.0",
"_name_or_path": "/media/damoxing/ckp/flowpro-pretrain/no4d-abl-abs-3500",
"action_dim": 30,
"action_head_dropout": 0.0,
"action_head_ffn_dim": 3072,
"action_head_hidden_size": 768,
"action_head_num_attention_heads": 12,
"action_head_num_layers": null,
"action_head_type": "shared",
"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
}