config.json

{
  "in_features": 512,
  "num_labels": 6
}