config.json

{
  "_name_or_path": "TuKoResearch/AuriStream",
  "architectures": ["AuriStream"],
  "model_type": "AuriStream.AuriStream",
  "auto_map": {
    "AutoConfig": "configuration_auristream.AuriStreamConfig",
    "AutoModel": "modeling_auristream.AuriStream",
    "AutoModelForCausalLM": "modeling_auristream.AuriStream"
  },
  "vocab_size": 8192,
  "n_embd": 768,
  "n_layer": 12,
  "n_head": 12,
  "dropout": 0.0,
  "bias": false,
  "use_rope": true,
  "skip_connections": true,
  "n_pred_steps": 20
}