| { | |
| "_class_name": "FSDPCausalWanNSAModel", | |
| "_diffusers_version": "0.37.1", | |
| "_name_or_path": "/mnt/localssd/omnimem-sf-t12-cd4500-h100-p2-20260925/assets/model_path/cd-step4500-ema", | |
| "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 | |
| } | |