ministral3b-neuron / neuron_config.json
kunhunjon's picture
Upload folder using huggingface_hub
7e2974f verified
{
"_serialized_key": "NxDNeuronConfig",
"batch_size": 1,
"capacity_factor": null,
"checkpoint_id": "mistralai/Ministral-3-3B-Reasoning-2512",
"checkpoint_revision": "039f888eb54340b5e9870721f3c249fbc809b8e8",
"continuous_batching": false,
"ep_degree": 1,
"fused_qkv": true,
"glu_mlp": true,
"local_ranks_size": 2,
"max_batch_size": 1,
"max_context_length": 4096,
"max_topk": 256,
"n_active_tokens": 4096,
"neuronxcc_version": "2.21.33363.0+82129205",
"on_device_sampling": true,
"optimum_neuron_version": "0.4.3.dev0",
"output_logits": false,
"pp_degree": 1,
"sequence_length": 4096,
"speculation_length": 0,
"start_rank_id": 0,
"target": "trn1",
"torch_dtype": "bfloat16",
"tp_degree": 2
}