config.json

{
  "architectures": ["TinyModel"],
  "model_type": "tiny",
  "vocab_size": 4096,
  "hidden_size": 128,
  "code_dim": 96,
  "num_hidden_layers": 6,
  "num_attention_heads": 8,
  "num_key_value_heads": 4,
  "max_position_embeddings": 2048,
  "tie_word_embeddings": true,
  "auto_map": {
    "AutoModel": "model_tiny.TinyModel",
    "AutoModelForCausalLM": "model_tiny.TinyModel"
  }
}