File size: 696 Bytes
3b8f9ea | 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 26 27 28 29 30 | {
"_class_name": "FSDPCausalWanNSAModel",
"_diffusers_version": "0.37.1",
"_name_or_path": "/mnt/localssd/omnimem-sf-t12-ode-h100-p2-20260926/assets/model_path/ode-step800-model",
"chunk_length": 4680,
"dim": 1536,
"enable_nsa": true,
"eps": 1e-06,
"exclude_sink_chunks": 0,
"exclude_window_chunks": 3,
"ffn_dim": 8960,
"freq_dim": 256,
"gate_mode": "sigmoid",
"group_q_size": 15,
"in_dim": 16,
"model_type": "t2v",
"num_heads": 12,
"num_kv_head_groups": null,
"num_layers": 30,
"num_selected_blocks": 12,
"out_dim": 16,
"pool_f": 1,
"pool_h": 15,
"pool_w": 2,
"progressive_exclude": true,
"skip_module": null,
"text_len": 512,
"tf": false
}
|