File size: 602 Bytes
5239120 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | {
"model_type": "AuriStream",
"architectures": [
"AuriStreamModel"
],
"auto_map": {
"AutoConfig": "TuKoResearch/AuriStream-base--configuration_auristream.AuriStreamConfig",
"AutoModel": "TuKoResearch/AuriStream-base--modeling_auristream.AuriStreamModel",
"AutoModelForCausalLM": "TuKoResearch/AuriStream-base--modeling_auristream.AuriStreamModel"
},
"torch_dtype": "bfloat16",
"transformers_version": "4.40.0",
"vocab_size": 8192,
"n_embd": 2560,
"n_layer": 96,
"n_head": 32,
"n_pred_steps": 40,
"dropout": 0.0,
"bias": false,
"input_conv_kernel_size": 0
} |