config.json

{
  "auto_map":{
    "AutoConfig": "model_configuration.ModelConfig" ,
    "AutoModel": "modelling.MyModel",
  }, 
  "dropout": 0.2,
  "embed_size": 256,
  "latent_dim": 16,
  "max_len": 128,
  "num_classes": 5,
  "num_heads": 2,
  "num_layers": 2,
  "vocab_size": 8192
}