config.json

{
  "architectures": [
    "LightningTransformerModel"
  ],
  "cfg": {
    "batch_size": 1,
    "block_num": 6,
    "embed_dims": 512,
    "head_size": 64,
    "iterations": 250,
    "lr": 0.0002,
    "num_heads": 8,
    "seq_len": 164,
    "use_liger": true,
    "vocab_size": 49152
  },
  "dtype": "float32",
  "model_type": "lightning_transformer",
  "transformers_version": "5.12.0"
}