variant.json

{
  "model_configuration": {
    "num_layers": 6,
    "hidden_size": 512,
    "num_heads": 8,
    "ffn_size": 2048
  },
  ...
}