File size: 716 Bytes
a01e851
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
{
  "_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
}