config.json

{
  "vocab_size": 1968,
  "output_size": 128,
  "embedding_dim": 256,
  "num_layers": 8,
  "num_heads": 8,
  "max_sequence_length": 79,
  "num_return_buckets": 128,
  "model_name": "9M"
}