config.json

{
    "hidden_size": 256,
    "num_hidden_layers": 9,
    "dropout": 0.0,
    "num_attention_heads": 4,
    "intermediate_size": 32,
    "model_type": "custom_model"
}