{
  "_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
}