config.json

{
  "n_layers": 2,
  "model_type": "my_model",
  "auto_map": {
    "AutoConfig": "configuration_my_model.MyModelConfig",
    "AutoModel": "modeling_my_model.MyModel",
    "TFAutoModel": "modeling_tf_my_model.TFMyModel"
  }
}