| { |
| "_class_name": "A2Model", |
| "_diffusers_version": "0.33.0.dev0", |
| "_name_or_path": "/maindata/data/shared/public/debang.li/codes/multi-subject-video/new_diffusers/train/SkyReels-C1/ckpts/20250330_wanx_i2v_29w_human_ratio_0_4/checkpoint-1300/transformer", |
| "added_kv_proj_dim": 5120, |
| "attention_head_dim": 128, |
| "cross_attn_norm": true, |
| "eps": 1e-06, |
| "ffn_dim": 13824, |
| "freq_dim": 256, |
| "image_dim": 1280, |
| "in_channels": 36, |
| "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 |
| } |
|
|