config.json

{
  "architectures": ["ZAI2"],
  "model_type": "flagship",
  "vocab_size": 50257,
  "hidden_size": 5120,
  "num_hidden_layers": 40,
  "auto_map": {
    "AutoConfig": "configuration_hyperion.HyperionConfig",
    "AutoModelForCausalLM": "modeling_hyperion.ZAI2Hyperion12B"
  }
}