{
  "embed_tokens.weight": {
    "offset": 0,
    "count": 16777216,
    "shape": [
      16384,
      1024
    ]
  },
  "pos_emb.weight": {
    "offset": 16777216,
    "count": 1048576,
    "shape": [
      1024,
      1024
    ]
  }
}