config.json

{
  "architectures": [
    "MinimalTransformer"
  ],
  "model_type": "custom",
  "embed_dim": 64,
  "hidden_dim": 128,
  "num_blocks": 2
}