config.json

{
  "n_layers": 2,
  "model_type": "my_model",
  "auto_map": {
    "AutoConfig": "configuration_my_model.MyModelConfig"
  }
}