config.json

{
  "auto_map": {
    "AutoConfig": "configuration_spin.SpinConfig",
    "AutoModelForCausalLM": "modeling_spin.SpinForCausalLM"
  },
  "d_ff": 128,
  "d_model": 48,
  "max_seq_len": 256,
  "model_type": "spin",
  "n_heads": 4,
  "n_layers": 2,
  "norm_eps": 1e-05,
  "transformers_version": "4.57.6",
  "vocab_size": 512
}