config.json

{
  "embed_size": 64,
  "embedding_type": "scaled",
  "k": 5,
  "num_inner_layers": 3,
  "vocab_size": 50257
}