|
{ |
|
"builder_config": { |
|
"fp8": false, |
|
"hidden_act": "silu", |
|
"hidden_size": 4096, |
|
"int8": true, |
|
"max_batch_size": 8, |
|
"max_input_len": 2048, |
|
"max_num_tokens": null, |
|
"max_output_len": 512, |
|
"max_position_embeddings": 4096, |
|
"name": "llama", |
|
"num_heads": 32, |
|
"num_kv_heads": 32, |
|
"num_layers": 32, |
|
"parallel_build": false, |
|
"pipeline_parallel": 1, |
|
"precision": "float16", |
|
"quant_mode": 70, |
|
"tensor_parallel": 8, |
|
"use_refit": false, |
|
"vocab_size": 32000 |
|
}, |
|
"plugin_config": { |
|
"attention_qk_half_accumulation": false, |
|
"bert_attention_plugin": false, |
|
"context_fmha_type": 0, |
|
"gemm_plugin": "float16", |
|
"gpt_attention_plugin": "float16", |
|
"identity_plugin": false, |
|
"layernorm_plugin": false, |
|
"layernorm_quantization_plugin": false, |
|
"lookup_plugin": false, |
|
"nccl_plugin": "float16", |
|
"paged_kv_cache": false, |
|
"quantize_per_token_plugin": true, |
|
"quantize_tensor_plugin": true, |
|
"remove_input_padding": true, |
|
"rmsnorm_plugin": false, |
|
"rmsnorm_quantization_plugin": "float16", |
|
"smooth_quant_gemm_plugin": "float16", |
|
"tokens_per_block": 0, |
|
"use_custom_all_reduce": false, |
|
"weight_only_groupwise_quant_matmul_plugin": false, |
|
"weight_only_quant_matmul_plugin": false |
|
} |
|
} |
|
|