{
  "format": "W4A16_NVFP4",
  "quantized_modules": 256,
  "modules": {
    "model.layers.0.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.0.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.0.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.0.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.0.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.0.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.1.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.1.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.1.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.1.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.1.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.1.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.2.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.2.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.2.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.2.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.2.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.2.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.3.self_attn.q_proj": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.3.self_attn.k_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.3.self_attn.v_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.3.self_attn.o_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.3.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.3.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.3.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.3.memory.q_proj": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.3.memory.k_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.3.memory.v_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.3.memory.o_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.3.memory.write_gate_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.3.memory.write_up_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.3.memory.write_down_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.4.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.4.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.4.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.4.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.4.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.4.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.5.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.5.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.5.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.5.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.5.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.5.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.6.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.6.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.6.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.6.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.6.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.6.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.7.self_attn.q_proj": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.7.self_attn.k_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.7.self_attn.v_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.7.self_attn.o_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.7.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.7.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.7.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.7.memory.q_proj": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.7.memory.k_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.7.memory.v_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.7.memory.o_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.7.memory.write_gate_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.7.memory.write_up_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.7.memory.write_down_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.8.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.8.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.8.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.8.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.8.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.8.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.9.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.9.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.9.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.9.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.9.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.9.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.10.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.10.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.10.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.10.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.10.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.10.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.11.self_attn.q_proj": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.11.self_attn.k_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.11.self_attn.v_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.11.self_attn.o_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.11.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.11.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.11.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.11.memory.q_proj": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.11.memory.k_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.11.memory.v_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.11.memory.o_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.11.memory.write_gate_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.11.memory.write_up_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.11.memory.write_down_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.12.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.12.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.12.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.12.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.12.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.12.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.13.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.13.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.13.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.13.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.13.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.13.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.14.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.14.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.14.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.14.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.14.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.14.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.15.self_attn.q_proj": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.15.self_attn.k_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.15.self_attn.v_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.15.self_attn.o_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.15.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.15.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.15.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.15.memory.q_proj": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.15.memory.k_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.15.memory.v_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.15.memory.o_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.15.memory.write_gate_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.15.memory.write_up_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.15.memory.write_down_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.16.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.16.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.16.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.16.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.16.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.16.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.17.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.17.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.17.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.17.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.17.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.17.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.18.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.18.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.18.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.18.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.18.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.18.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.19.self_attn.q_proj": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.19.self_attn.k_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.19.self_attn.v_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.19.self_attn.o_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.19.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.19.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.19.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.19.memory.q_proj": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.19.memory.k_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.19.memory.v_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.19.memory.o_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.19.memory.write_gate_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.19.memory.write_up_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.19.memory.write_down_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.20.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.20.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.20.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.20.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.20.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.20.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.21.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.21.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.21.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.21.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.21.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.21.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.22.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.22.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.22.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.22.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.22.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.22.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.23.self_attn.q_proj": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.23.self_attn.k_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.23.self_attn.v_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.23.self_attn.o_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.23.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.23.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.23.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.23.memory.q_proj": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.23.memory.k_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.23.memory.v_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.23.memory.o_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.23.memory.write_gate_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.23.memory.write_up_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.23.memory.write_down_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.24.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.24.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.24.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.24.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.24.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.24.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.25.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.25.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.25.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.25.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.25.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.25.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.26.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.26.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.26.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.26.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.26.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.26.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.27.self_attn.q_proj": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.27.self_attn.k_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.27.self_attn.v_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.27.self_attn.o_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.27.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.27.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.27.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.27.memory.q_proj": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.27.memory.k_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.27.memory.v_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.27.memory.o_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.27.memory.write_gate_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.27.memory.write_up_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.27.memory.write_down_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.28.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.28.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.28.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.28.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.28.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.28.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.29.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.29.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.29.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.29.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.29.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.29.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.30.linear_attn.out_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.30.linear_attn.in_proj_qkv": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.30.linear_attn.in_proj_z": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.30.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.30.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.30.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.31.self_attn.q_proj": {
      "shape": [
        8192,
        2560
      ],
      "weight_bytes": 10485760,
      "scale_bytes": 1310724
    },
    "model.layers.31.self_attn.k_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.31.self_attn.v_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.31.self_attn.o_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.31.mlp.gate_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.31.mlp.up_proj": {
      "shape": [
        9216,
        2560
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.31.mlp.down_proj": {
      "shape": [
        2560,
        9216
      ],
      "weight_bytes": 11796480,
      "scale_bytes": 1474564
    },
    "model.layers.31.memory.q_proj": {
      "shape": [
        4096,
        2560
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.31.memory.k_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.31.memory.v_proj": {
      "shape": [
        1024,
        2560
      ],
      "weight_bytes": 1310720,
      "scale_bytes": 163844
    },
    "model.layers.31.memory.o_proj": {
      "shape": [
        2560,
        4096
      ],
      "weight_bytes": 5242880,
      "scale_bytes": 655364
    },
    "model.layers.31.memory.write_gate_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.31.memory.write_up_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    },
    "model.layers.31.memory.write_down_proj": {
      "shape": [
        2560,
        2560
      ],
      "weight_bytes": 3276800,
      "scale_bytes": 409604
    }
  },
  "conversion_seconds": 50.26564243901521,
  "weight_bytes": 3503765184,
  "gpu": "NVIDIA H800",
  "modelopt_version": "0.47.0"
}