config.json

{
  "architectures": [
    "LightningTransformerModel"
  ],
  "auto_map": {
    "AutoConfig": "configuration_lightningtransformer.LightningTransformerModelConfig",
    "AutoModel": "modeling_lightningtransformer.LightningTransformerModel",
    "AutoModelForCausalLM": "modeling_lightningtransformer.LightningTransformerModelForCausalLM"
  },
  "cfg": {
    "batch_size": 1,
    "block_num": 1,
    "embed_dims": 4,
    "head_size": 4,
    "iterations": 10,
    "lr": 0.00025,
    "num_heads": 1,
    "seq_len": 1,
    "use_liger": false,
    "vocab_size": 49152
  },
  "dtype": "float32",
  "model_type": "lightning_transformer",
  "num_hidden_layers": 4,
  "transformers_version": "5.12.1"
}