Loi-LLM / Neuron /neuron_config.json
TensorVizion's picture
Upload 4 files
a01e851 verified
Raw
History Blame Contribute Delete
716 Bytes
{
"_serialized_key": "NxDNeuronConfig",
"batch_size": 4,
"capacity_factor": null,
"checkpoint_id": "TensorVizion/Loi-LLM",
"checkpoint_revision": "08551e259be6cc7aa351aa05af997d309e7ffaa2",
"continuous_batching": true,
"ep_degree": 1,
"fused_qkv": true,
"glu_mlp": true,
"local_ranks_size": 4,
"max_batch_size": 4,
"max_context_length": 4096,
"max_topk": 256,
"n_active_tokens": 4096,
"neuronxcc_version": "2.21.18209.0+043b1bf7",
"on_device_sampling": true,
"optimum_neuron_version": "0.4.1",
"output_logits": false,
"pp_degree": 1,
"sequence_length": 4096,
"speculation_length": 0,
"start_rank_id": 0,
"target": "inf2",
"torch_dtype": "bfloat16",
"tp_degree": 4
}