Brain2nd's picture
Fix pretrain EOS ids
e1a87fd verified
{
"D": 1024,
"D_ff": 3072,
"D_key": 128,
"D_value": 128,
"K": 12,
"N": 8,
"ahp_init": 0.0,
"architectures": [
"NeuronSparkForCausalLM"
],
"auto_map": {
"AutoConfig": "configuration_neuronspark.NeuronSparkConfig",
"AutoModelForCausalLM": "modeling_neuronspark.NeuronSparkForCausalLM"
},
"bias_balancing_ema": 0.99,
"bias_balancing_lr": 0.001,
"bos_token_id": null,
"dtype": "bfloat16",
"eos_token_id": 128361,
"eps_explore": 0.05,
"k_predictor_hidden": 256,
"memory_layer_interval": 4,
"model_type": "neuronspark",
"num_hidden_layers": 24,
"num_layers": 24,
"pad_token_id": 128361,
"ponder_T_final": 0.3,
"ponder_T_init": 2.0,
"rope_layout": "transformer_interleaved",
"spike_mode": "quantal",
"surrogate_alpha": 4.0,
"transformers_version": "5.6.2",
"use_ahp": false,
"use_cache": false,
"v_th_min": 0.02,
"v_th_reg_weight": 0.0,
"vocab_size": 128387
}