klemenk's picture
Update config.json
5e1117b verified
{
"_name_or_path": "TuKoResearch/AuriStream1B_60Pred_librilight_500k",
"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": 1280,
"n_layer": 48,
"n_head": 16,
"dropout": 0.0,
"bias": false,
"use_rope": false,
"n_pred_steps": 60,
"seq_len": 4096
}