aws-neuron/Qwen3-1.7B-TP4-BS4-SEQ2048
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

初始化项目，由ModelHub XC社区提供模型

Browse Source 
Model: aws-neuron/Qwen3-1.7B-TP4-BS4-SEQ2048
Source: Original Platform
This commit is contained in:
ModelHub XC
2026-08-30 21:56:23 +08:00
commit fd57cdfd17
114 changed files with 237628 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
context_encoding_model/_tp0_bk0/command.txt Normal file
View File

@@ -0,0 +1 @@
neuronx-cc compile --framework=XLA model.MODULE_bdcbea8455ebae357b4c+541d7181.hlo_module.pb --output model.MODULE_bdcbea8455ebae357b4c+541d7181.neff --target=trn2 --auto-cast=none --model-type=transformer '--tensorizer-options=--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-strided-dma ' --lnc=2 -O1 '--internal-hlo2tensorizer-options= --modular-flow-mac-threshold=10 --verify-hlo=true' --logfile=log-neuron-cc.txt --verbose=35

1
context_encoding_model/_tp0_bk0/compile_flags.MODULE_bdcbea8455ebae357b4c+541d7181.json Normal file
View File

@@ -0,0 +1 @@
["--target=trn2", "--auto-cast=none", "--model-type=transformer", "--tensorizer-options=--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-strided-dma ", "--lnc=2", "-O1", "--internal-hlo2tensorizer-options= --modular-flow-mac-threshold=10 --verify-hlo=true", "--logfile=/models/Qwen3-1.7B-compiled/context_encoding_model/_tp0_bk0/log-neuron-cc.txt"]

1147
context_encoding_model/_tp0_bk0/global_metric_store.json Normal file
View File

File diff suppressed because it is too large Load Diff

3
context_encoding_model/_tp0_bk0/graph.neff Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:92d81f134ce62a3b1bcb9c577d19d5ea4c462cc6a79b0b0e499d1560e903979b
size 830464

9211
context_encoding_model/_tp0_bk0/log-neuron-cc.txt Normal file
View File

File diff suppressed because it is too large Load Diff

3
context_encoding_model/_tp0_bk0/metaneff.pb Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:ec4b0c42828e50013dc0bd682bdb81c0aa36920c9456fd53a653c18581ad49d1
size 1877151

3
context_encoding_model/_tp0_bk0/model.MODULE_bdcbea8455ebae357b4c+541d7181.hlo_module.pb Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:5239d1061207e7bd19a6be27bb12fddcd4b60b66cb5ba20fb4ab47a4be30d4c5
size 1961858

3
context_encoding_model/_tp0_bk0/model.MODULE_bdcbea8455ebae357b4c+541d7181.neff Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:92d81f134ce62a3b1bcb9c577d19d5ea4c462cc6a79b0b0e499d1560e903979b
size 830464

224
context_encoding_model/_tp0_bk0/neuron_config.json Normal file
View File

@@ -0,0 +1,224 @@
{
"_attn_implementation_autoset": false,
"_name_or_path": "/models/Qwen3-1.7B/",
"add_cross_attention": false,
"architectures": [
"Qwen3ForCausalLM"
],
"attention_bias": false,
"attention_dropout": 0.0,
"attribute_map": {},
"bad_words_ids": null,
"begin_suppress_tokens": null,
"bos_token_id": 151643,
"chunk_size_feed_forward": 0,
"cross_attention_hidden_size": null,
"decoder_start_token_id": null,
"diversity_penalty": 0.0,
"do_sample": false,
"early_stopping": false,
"encoder_no_repeat_ngram_size": 0,
"eos_token_id": 151645,
"exponential_decay_length_penalty": null,
"finetuning_task": null,
"forced_bos_token_id": null,
"forced_eos_token_id": null,
"fused_spec_config": null,
"head_dim": 128,
"hidden_act": "silu",
"hidden_size": 2048,
"id2label": {
"0": "LABEL_0",
"1": "LABEL_1"
},
"initializer_range": 0.02,
"intermediate_size": 6144,
"is_decoder": false,
"is_encoder_decoder": false,
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1
},
"length_penalty": 1.0,
"max_length": 20,
"max_position_embeddings": 40960,
"max_window_layers": 28,
"metadata": null,
"min_length": 0,
"model_type": "qwen3",
"neuron_config": {
"activation_quantization_type": null,
"allow_input_truncation": false,
"apply_seq_ids_mask": false,
"async_mode": false,
"attention_dp_degree": 1,
"attention_dtype": null,
"attn_block_cte_nki_kernel_enabled": false,
"attn_block_tkg_nki_kernel_cache_update": false,
"attn_block_tkg_nki_kernel_cascaded_attention": false,
"attn_block_tkg_nki_kernel_enabled": false,
"attn_cls": {
"__module__": "neuronx_distributed_inference.models.qwen3.modeling_qwen3",
"__name__": "NeuronQwen3Attention"
},
"attn_kernel_enabled": null,
"attn_tkg_builtin_kernel_enabled": false,
"attn_tkg_nki_kernel_enabled": false,
"batch_size": 1,
"bucket_n_active_tokens": true,
"buckets": [
128
],
"cast_type": "config",
"cc_pipeline_tiling_factor": 2,
"chunked_prefill_config": null,
"context_encoding_buckets": [
128
],
"cp_degree": 1,
"ctx_batch_size": 1,
"disable_kv_cache_tiling": false,
"draft_model_modules_to_not_convert": null,
"enable_bucketing": true,
"enable_cte_modular_flow": false,
"enable_eagle_draft_input_norm": false,
"enable_eagle_speculation": false,
"enable_fused_speculation": false,
"enable_long_context_mode": false,
"enable_output_completion_notifications": false,
"enable_spill_reload_dge": false,
"enable_token_tree": false,
"ep_degree": 1,
"expert_mlp_nki_kernel_enabled": null,
"flash_decoding_enabled": false,
"fused_qkv": false,
"fused_rmsnorm_skip_gamma": false,
"is_block_kv_layout": null,
"is_chunked_prefill": false,
"is_continuous_batching": true,
"is_eagle_draft": false,
"is_medusa": false,
"is_prefill_stage": true,
"is_prefix_caching": false,
"k_cache_transposed": false,
"kv_cache_batch_size": 4,
"kv_cache_padding_size": 0,
"kv_cache_quant": false,
"kv_cache_tiling": false,
"layer_boundary_markers": false,
"lm_head_pad": true,
"lm_head_pad_alignment_size": 1,
"local_ranks_size": 4,
"logical_nc_config": 2,
"lora_config": null,
"max_batch_size": 4,
"max_context_length": 2048,
"max_length": 2048,
"max_new_tokens": null,
"medusa_speculation_length": 0,
"medusa_tree": null,
"mlp_kernel_enabled": false,
"mlp_kernel_fuse_residual_add": false,
"modules_to_not_convert": null,
"moe_fused_nki_kernel_enabled": null,
"n_active_tokens": 2048,
"n_positions": 2048,
"num_medusa_heads": 0,
"on_cpu": false,
"on_device_sampling_config": {
"deterministic": false,
"do_sample": false,
"dynamic": true,
"global_topk": 256,
"on_device_sampling_config": true,
"temperature": 1.0,
"top_k": 1,
"top_k_kernel_enabled": false,
"top_p": 1.0
},
"output_logits": false,
"overrides_torch_dtype": true,
"pa_block_size": 2048,
"pa_num_blocks": 4,
"padding_side": "right",
"pp_degree": 1,
"prefix_buckets": null,
"qk_layernorm": false,
"qkv_kernel_enabled": false,
"qkv_kernel_fuse_residual_add": false,
"qkv_kernel_nbsd_layout": false,
"quantization_dtype": "int8",
"quantization_type": "per_tensor_symmetric",
"quantize_clamp_bound": Infinity,
"quantized": false,
"quantized_checkpoints_path": null,
"quantized_mlp_kernel_enabled": false,
"rmsnorm_quantize_kernel_enabled": false,
"router_topk_nki_kernel_enabled": null,
"rpl_reduce_dtype": null,
"save_sharded_checkpoint": true,
"scratchpad_page_size": null,
"seq_len": 2048,
"seq_len_threshold_for_cc_tiling": 16384,
"sequence_parallel_enabled": false,
"shared_mlp_nki_kernel_enabled": null,
"skip_sharding": false,
"skip_warmup": false,
"spec_batch_size": 4,
"speculation_length": 0,
"start_rank_id": 0,
"strided_context_parallel_kernel_enabled": false,
"target": null,
"tensor_capture_config": null,
"tile_cc": false,
"tkg_batch_size": 4,
"token_generation_buckets": null,
"token_tree_config": null,
"torch_dtype": "bfloat16",
"tp_degree": 4,
"vocab_parallel": false,
"weight_gather_seq_len_threshold": 32768,
"weights_to_skip_layout_optimization": [],
"world_size": 4
},
"no_repeat_ngram_size": 0,
"num_attention_heads": 16,
"num_beam_groups": 1,
"num_beams": 1,
"num_cores_per_group": 1,
"num_hidden_layers": 28,
"num_key_value_heads": 8,
"num_return_sequences": 1,
"output_attentions": false,
"output_hidden_states": false,
"output_scores": false,
"pad_token_id": 0,
"prefix": null,
"problem_type": null,
"pruned_heads": {},
"remove_invalid_values": false,
"repetition_penalty": 1.0,
"return_dict": true,
"return_dict_in_generate": false,
"rms_norm_eps": 1e-06,
"rope_scaling": null,
"rope_theta": 1000000,
"sep_token_id": null,
"sliding_window": null,
"suppress_tokens": null,
"task_specific_params": null,
"temperature": 1.0,
"tf_legacy_loss": false,
"tie_encoder_decoder": false,
"tie_word_embeddings": true,
"tokenizer_class": null,
"top_k": 50,
"top_p": 1.0,
"torchscript": false,
"transformers_version": "4.51.0",
"typical_p": 1.0,
"use_bfloat16": false,
"use_cache": true,
"use_sliding_window": false,
"vocab_size": 151936
}

1
context_encoding_model/_tp0_bk1/command.txt Normal file
View File

@@ -0,0 +1 @@
neuronx-cc compile --framework=XLA model.MODULE_e7dad336ed1c266a3016+bbc3fa47.hlo_module.pb --output model.MODULE_e7dad336ed1c266a3016+bbc3fa47.neff --target=trn2 --auto-cast=none --model-type=transformer '--tensorizer-options=--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-strided-dma ' --lnc=2 -O1 '--internal-hlo2tensorizer-options= --modular-flow-mac-threshold=10 --verify-hlo=true' --logfile=log-neuron-cc.txt --verbose=35

1
context_encoding_model/_tp0_bk1/compile_flags.MODULE_e7dad336ed1c266a3016+bbc3fa47.json Normal file
View File

@@ -0,0 +1 @@
["--target=trn2", "--auto-cast=none", "--model-type=transformer", "--tensorizer-options=--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-strided-dma ", "--lnc=2", "-O1", "--internal-hlo2tensorizer-options= --modular-flow-mac-threshold=10 --verify-hlo=true", "--logfile=/models/Qwen3-1.7B-compiled/context_encoding_model/_tp0_bk1/log-neuron-cc.txt"]

1177
context_encoding_model/_tp0_bk1/global_metric_store.json Normal file
View File

File diff suppressed because it is too large Load Diff

3
context_encoding_model/_tp0_bk1/graph.neff Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:ca594c9d5f303adb5de3e5aa58b2eca9ac1ed3aff762e1b09620d6474407d8bc
size 861184

9530
context_encoding_model/_tp0_bk1/log-neuron-cc.txt Normal file
View File

File diff suppressed because it is too large Load Diff

3
context_encoding_model/_tp0_bk1/metaneff.pb Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:20425c1f386fac6326c8423f6243a37818ce0b36840edfe0fdf47dc9a74af5f2
size 2194530

3
context_encoding_model/_tp0_bk1/model.MODULE_e7dad336ed1c266a3016+bbc3fa47.hlo_module.pb Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:4dbd208cf8cb5cfeef58be408e28cdbf0a23d70a7cd3affd1510ceae9fb95c8b
size 2280924

3
context_encoding_model/_tp0_bk1/model.MODULE_e7dad336ed1c266a3016+bbc3fa47.neff Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:ca594c9d5f303adb5de3e5aa58b2eca9ac1ed3aff762e1b09620d6474407d8bc
size 861184

224
context_encoding_model/_tp0_bk1/neuron_config.json Normal file
View File

@@ -0,0 +1,224 @@
{
"_attn_implementation_autoset": false,
"_name_or_path": "/models/Qwen3-1.7B/",
"add_cross_attention": false,
"architectures": [
"Qwen3ForCausalLM"
],
"attention_bias": false,
"attention_dropout": 0.0,
"attribute_map": {},
"bad_words_ids": null,
"begin_suppress_tokens": null,
"bos_token_id": 151643,
"chunk_size_feed_forward": 0,
"cross_attention_hidden_size": null,
"decoder_start_token_id": null,
"diversity_penalty": 0.0,
"do_sample": false,
"early_stopping": false,
"encoder_no_repeat_ngram_size": 0,
"eos_token_id": 151645,
"exponential_decay_length_penalty": null,
"finetuning_task": null,
"forced_bos_token_id": null,
"forced_eos_token_id": null,
"fused_spec_config": null,
"head_dim": 128,
"hidden_act": "silu",
"hidden_size": 2048,
"id2label": {
"0": "LABEL_0",
"1": "LABEL_1"
},
"initializer_range": 0.02,
"intermediate_size": 6144,
"is_decoder": false,
"is_encoder_decoder": false,
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1
},
"length_penalty": 1.0,
"max_length": 20,
"max_position_embeddings": 40960,
"max_window_layers": 28,
"metadata": null,
"min_length": 0,
"model_type": "qwen3",
"neuron_config": {
"activation_quantization_type": null,
"allow_input_truncation": false,
"apply_seq_ids_mask": false,
"async_mode": false,
"attention_dp_degree": 1,
"attention_dtype": null,
"attn_block_cte_nki_kernel_enabled": false,
"attn_block_tkg_nki_kernel_cache_update": false,
"attn_block_tkg_nki_kernel_cascaded_attention": false,
"attn_block_tkg_nki_kernel_enabled": false,
"attn_cls": {
"__module__": "neuronx_distributed_inference.models.qwen3.modeling_qwen3",
"__name__": "NeuronQwen3Attention"
},
"attn_kernel_enabled": null,
"attn_tkg_builtin_kernel_enabled": false,
"attn_tkg_nki_kernel_enabled": false,
"batch_size": 1,
"bucket_n_active_tokens": true,
"buckets": [
256
],
"cast_type": "config",
"cc_pipeline_tiling_factor": 2,
"chunked_prefill_config": null,
"context_encoding_buckets": [
256
],
"cp_degree": 1,
"ctx_batch_size": 1,
"disable_kv_cache_tiling": false,
"draft_model_modules_to_not_convert": null,
"enable_bucketing": true,
"enable_cte_modular_flow": false,
"enable_eagle_draft_input_norm": false,
"enable_eagle_speculation": false,
"enable_fused_speculation": false,
"enable_long_context_mode": false,
"enable_output_completion_notifications": false,
"enable_spill_reload_dge": false,
"enable_token_tree": false,
"ep_degree": 1,
"expert_mlp_nki_kernel_enabled": null,
"flash_decoding_enabled": false,
"fused_qkv": false,
"fused_rmsnorm_skip_gamma": false,
"is_block_kv_layout": null,
"is_chunked_prefill": false,
"is_continuous_batching": true,
"is_eagle_draft": false,
"is_medusa": false,
"is_prefill_stage": true,
"is_prefix_caching": false,
"k_cache_transposed": false,
"kv_cache_batch_size": 4,
"kv_cache_padding_size": 0,
"kv_cache_quant": false,
"kv_cache_tiling": false,
"layer_boundary_markers": false,
"lm_head_pad": true,
"lm_head_pad_alignment_size": 1,
"local_ranks_size": 4,
"logical_nc_config": 2,
"lora_config": null,
"max_batch_size": 4,
"max_context_length": 2048,
"max_length": 2048,
"max_new_tokens": null,
"medusa_speculation_length": 0,
"medusa_tree": null,
"mlp_kernel_enabled": false,
"mlp_kernel_fuse_residual_add": false,
"modules_to_not_convert": null,
"moe_fused_nki_kernel_enabled": null,
"n_active_tokens": 2048,
"n_positions": 2048,
"num_medusa_heads": 0,
"on_cpu": false,
"on_device_sampling_config": {
"deterministic": false,
"do_sample": false,
"dynamic": true,
"global_topk": 256,
"on_device_sampling_config": true,
"temperature": 1.0,
"top_k": 1,
"top_k_kernel_enabled": false,
"top_p": 1.0
},
"output_logits": false,
"overrides_torch_dtype": true,
"pa_block_size": 2048,
"pa_num_blocks": 4,
"padding_side": "right",
"pp_degree": 1,
"prefix_buckets": null,
"qk_layernorm": false,
"qkv_kernel_enabled": false,
"qkv_kernel_fuse_residual_add": false,
"qkv_kernel_nbsd_layout": false,
"quantization_dtype": "int8",
"quantization_type": "per_tensor_symmetric",
"quantize_clamp_bound": Infinity,
"quantized": false,
"quantized_checkpoints_path": null,
"quantized_mlp_kernel_enabled": false,
"rmsnorm_quantize_kernel_enabled": false,
"router_topk_nki_kernel_enabled": null,
"rpl_reduce_dtype": null,
"save_sharded_checkpoint": true,
"scratchpad_page_size": null,
"seq_len": 2048,
"seq_len_threshold_for_cc_tiling": 16384,
"sequence_parallel_enabled": false,
"shared_mlp_nki_kernel_enabled": null,
"skip_sharding": false,
"skip_warmup": false,
"spec_batch_size": 4,
"speculation_length": 0,
"start_rank_id": 0,
"strided_context_parallel_kernel_enabled": false,
"target": null,
"tensor_capture_config": null,
"tile_cc": false,
"tkg_batch_size": 4,
"token_generation_buckets": null,
"token_tree_config": null,
"torch_dtype": "bfloat16",
"tp_degree": 4,
"vocab_parallel": false,
"weight_gather_seq_len_threshold": 32768,
"weights_to_skip_layout_optimization": [],
"world_size": 4
},
"no_repeat_ngram_size": 0,
"num_attention_heads": 16,
"num_beam_groups": 1,
"num_beams": 1,
"num_cores_per_group": 1,
"num_hidden_layers": 28,
"num_key_value_heads": 8,
"num_return_sequences": 1,
"output_attentions": false,
"output_hidden_states": false,
"output_scores": false,
"pad_token_id": 0,
"prefix": null,
"problem_type": null,
"pruned_heads": {},
"remove_invalid_values": false,
"repetition_penalty": 1.0,
"return_dict": true,
"return_dict_in_generate": false,
"rms_norm_eps": 1e-06,
"rope_scaling": null,
"rope_theta": 1000000,
"sep_token_id": null,
"sliding_window": null,
"suppress_tokens": null,
"task_specific_params": null,
"temperature": 1.0,
"tf_legacy_loss": false,
"tie_encoder_decoder": false,
"tie_word_embeddings": true,
"tokenizer_class": null,
"top_k": 50,
"top_p": 1.0,
"torchscript": false,
"transformers_version": "4.51.0",
"typical_p": 1.0,
"use_bfloat16": false,
"use_cache": true,
"use_sliding_window": false,
"vocab_size": 151936
}

1
context_encoding_model/_tp0_bk2/command.txt Normal file
View File

@@ -0,0 +1 @@
neuronx-cc compile --framework=XLA model.MODULE_f36c9ad51e28c98c9723+c4081b94.hlo_module.pb --output model.MODULE_f36c9ad51e28c98c9723+c4081b94.neff --target=trn2 --auto-cast=none --model-type=transformer '--tensorizer-options=--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-strided-dma ' --lnc=2 -O1 '--internal-hlo2tensorizer-options= --modular-flow-mac-threshold=10 --verify-hlo=true' --logfile=log-neuron-cc.txt --verbose=35

1
context_encoding_model/_tp0_bk2/compile_flags.MODULE_f36c9ad51e28c98c9723+c4081b94.json Normal file
View File

@@ -0,0 +1 @@
["--target=trn2", "--auto-cast=none", "--model-type=transformer", "--tensorizer-options=--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-strided-dma ", "--lnc=2", "-O1", "--internal-hlo2tensorizer-options= --modular-flow-mac-threshold=10 --verify-hlo=true", "--logfile=/models/Qwen3-1.7B-compiled/context_encoding_model/_tp0_bk2/log-neuron-cc.txt"]

1177
context_encoding_model/_tp0_bk2/global_metric_store.json Normal file
View File

File diff suppressed because it is too large Load Diff

3
context_encoding_model/_tp0_bk2/graph.neff Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:f7292240129556fa9d8d8412b12947fe2577cd0c6367f047d23e7e4526896cce
size 912384

9521
context_encoding_model/_tp0_bk2/log-neuron-cc.txt Normal file
View File

File diff suppressed because it is too large Load Diff

3
context_encoding_model/_tp0_bk2/metaneff.pb Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:3a52b69e5f1f8c87121e293c72dfb7fb9dc3f8050a04452be5e7691addcc3371
size 2280546

3
context_encoding_model/_tp0_bk2/model.MODULE_f36c9ad51e28c98c9723+c4081b94.hlo_module.pb Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:89ed0e9df247f2125a989d4e0bbc25ce61ee24d187117a143bd2bd583a741239
size 2366940

3
context_encoding_model/_tp0_bk2/model.MODULE_f36c9ad51e28c98c9723+c4081b94.neff Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:f7292240129556fa9d8d8412b12947fe2577cd0c6367f047d23e7e4526896cce
size 912384

224
context_encoding_model/_tp0_bk2/neuron_config.json Normal file
View File

@@ -0,0 +1,224 @@
{
"_attn_implementation_autoset": false,
"_name_or_path": "/models/Qwen3-1.7B/",
"add_cross_attention": false,
"architectures": [
"Qwen3ForCausalLM"
],
"attention_bias": false,
"attention_dropout": 0.0,
"attribute_map": {},
"bad_words_ids": null,
"begin_suppress_tokens": null,
"bos_token_id": 151643,
"chunk_size_feed_forward": 0,
"cross_attention_hidden_size": null,
"decoder_start_token_id": null,
"diversity_penalty": 0.0,
"do_sample": false,
"early_stopping": false,
"encoder_no_repeat_ngram_size": 0,
"eos_token_id": 151645,
"exponential_decay_length_penalty": null,
"finetuning_task": null,
"forced_bos_token_id": null,
"forced_eos_token_id": null,
"fused_spec_config": null,
"head_dim": 128,
"hidden_act": "silu",
"hidden_size": 2048,
"id2label": {
"0": "LABEL_0",
"1": "LABEL_1"
},
"initializer_range": 0.02,
"intermediate_size": 6144,
"is_decoder": false,
"is_encoder_decoder": false,
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1
},
"length_penalty": 1.0,
"max_length": 20,
"max_position_embeddings": 40960,
"max_window_layers": 28,
"metadata": null,
"min_length": 0,
"model_type": "qwen3",
"neuron_config": {
"activation_quantization_type": null,
"allow_input_truncation": false,
"apply_seq_ids_mask": false,
"async_mode": false,
"attention_dp_degree": 1,
"attention_dtype": null,
"attn_block_cte_nki_kernel_enabled": false,
"attn_block_tkg_nki_kernel_cache_update": false,
"attn_block_tkg_nki_kernel_cascaded_attention": false,
"attn_block_tkg_nki_kernel_enabled": false,
"attn_cls": {
"__module__": "neuronx_distributed_inference.models.qwen3.modeling_qwen3",
"__name__": "NeuronQwen3Attention"
},
"attn_kernel_enabled": null,
"attn_tkg_builtin_kernel_enabled": false,
"attn_tkg_nki_kernel_enabled": false,
"batch_size": 1,
"bucket_n_active_tokens": true,
"buckets": [
512
],
"cast_type": "config",
"cc_pipeline_tiling_factor": 2,
"chunked_prefill_config": null,
"context_encoding_buckets": [
512
],
"cp_degree": 1,
"ctx_batch_size": 1,
"disable_kv_cache_tiling": false,
"draft_model_modules_to_not_convert": null,
"enable_bucketing": true,
"enable_cte_modular_flow": false,
"enable_eagle_draft_input_norm": false,
"enable_eagle_speculation": false,
"enable_fused_speculation": false,
"enable_long_context_mode": false,
"enable_output_completion_notifications": false,
"enable_spill_reload_dge": false,
"enable_token_tree": false,
"ep_degree": 1,
"expert_mlp_nki_kernel_enabled": null,
"flash_decoding_enabled": false,
"fused_qkv": false,
"fused_rmsnorm_skip_gamma": false,
"is_block_kv_layout": null,
"is_chunked_prefill": false,
"is_continuous_batching": true,
"is_eagle_draft": false,
"is_medusa": false,
"is_prefill_stage": true,
"is_prefix_caching": false,
"k_cache_transposed": false,
"kv_cache_batch_size": 4,
"kv_cache_padding_size": 0,
"kv_cache_quant": false,
"kv_cache_tiling": false,
"layer_boundary_markers": false,
"lm_head_pad": true,
"lm_head_pad_alignment_size": 1,
"local_ranks_size": 4,
"logical_nc_config": 2,
"lora_config": null,
"max_batch_size": 4,
"max_context_length": 2048,
"max_length": 2048,
"max_new_tokens": null,
"medusa_speculation_length": 0,
"medusa_tree": null,
"mlp_kernel_enabled": false,
"mlp_kernel_fuse_residual_add": false,
"modules_to_not_convert": null,
"moe_fused_nki_kernel_enabled": null,
"n_active_tokens": 2048,
"n_positions": 2048,
"num_medusa_heads": 0,
"on_cpu": false,
"on_device_sampling_config": {
"deterministic": false,
"do_sample": false,
"dynamic": true,
"global_topk": 256,
"on_device_sampling_config": true,
"temperature": 1.0,
"top_k": 1,
"top_k_kernel_enabled": false,
"top_p": 1.0
},
"output_logits": false,
"overrides_torch_dtype": true,
"pa_block_size": 2048,
"pa_num_blocks": 4,
"padding_side": "right",
"pp_degree": 1,
"prefix_buckets": null,
"qk_layernorm": false,
"qkv_kernel_enabled": false,
"qkv_kernel_fuse_residual_add": false,
"qkv_kernel_nbsd_layout": false,
"quantization_dtype": "int8",
"quantization_type": "per_tensor_symmetric",
"quantize_clamp_bound": Infinity,
"quantized": false,
"quantized_checkpoints_path": null,
"quantized_mlp_kernel_enabled": false,
"rmsnorm_quantize_kernel_enabled": false,
"router_topk_nki_kernel_enabled": null,
"rpl_reduce_dtype": null,
"save_sharded_checkpoint": true,
"scratchpad_page_size": null,
"seq_len": 2048,
"seq_len_threshold_for_cc_tiling": 16384,
"sequence_parallel_enabled": false,
"shared_mlp_nki_kernel_enabled": null,
"skip_sharding": false,
"skip_warmup": false,
"spec_batch_size": 4,
"speculation_length": 0,
"start_rank_id": 0,
"strided_context_parallel_kernel_enabled": false,
"target": null,
"tensor_capture_config": null,
"tile_cc": false,
"tkg_batch_size": 4,
"token_generation_buckets": null,
"token_tree_config": null,
"torch_dtype": "bfloat16",
"tp_degree": 4,
"vocab_parallel": false,
"weight_gather_seq_len_threshold": 32768,
"weights_to_skip_layout_optimization": [],
"world_size": 4
},
"no_repeat_ngram_size": 0,
"num_attention_heads": 16,
"num_beam_groups": 1,
"num_beams": 1,
"num_cores_per_group": 1,
"num_hidden_layers": 28,
"num_key_value_heads": 8,
"num_return_sequences": 1,
"output_attentions": false,
"output_hidden_states": false,
"output_scores": false,
"pad_token_id": 0,
"prefix": null,
"problem_type": null,
"pruned_heads": {},
"remove_invalid_values": false,
"repetition_penalty": 1.0,
"return_dict": true,
"return_dict_in_generate": false,
"rms_norm_eps": 1e-06,
"rope_scaling": null,
"rope_theta": 1000000,
"sep_token_id": null,
"sliding_window": null,
"suppress_tokens": null,
"task_specific_params": null,
"temperature": 1.0,
"tf_legacy_loss": false,
"tie_encoder_decoder": false,
"tie_word_embeddings": true,
"tokenizer_class": null,
"top_k": 50,
"top_p": 1.0,
"torchscript": false,
"transformers_version": "4.51.0",
"typical_p": 1.0,
"use_bfloat16": false,
"use_cache": true,
"use_sliding_window": false,
"vocab_size": 151936
}

1
context_encoding_model/_tp0_bk3/command.txt Normal file
View File

@@ -0,0 +1 @@
neuronx-cc compile --framework=XLA model.MODULE_0984f4c19a044cc11c2a+1c024d2c.hlo_module.pb --output model.MODULE_0984f4c19a044cc11c2a+1c024d2c.neff --target=trn2 --auto-cast=none --model-type=transformer '--tensorizer-options=--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-strided-dma ' --lnc=2 -O1 '--internal-hlo2tensorizer-options= --modular-flow-mac-threshold=10 --verify-hlo=true' --logfile=log-neuron-cc.txt --verbose=35

1
context_encoding_model/_tp0_bk3/compile_flags.MODULE_0984f4c19a044cc11c2a+1c024d2c.json Normal file
View File

@@ -0,0 +1 @@
["--target=trn2", "--auto-cast=none", "--model-type=transformer", "--tensorizer-options=--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-strided-dma ", "--lnc=2", "-O1", "--internal-hlo2tensorizer-options= --modular-flow-mac-threshold=10 --verify-hlo=true", "--logfile=/models/Qwen3-1.7B-compiled/context_encoding_model/_tp0_bk3/log-neuron-cc.txt"]

1177
context_encoding_model/_tp0_bk3/global_metric_store.json Normal file
View File

File diff suppressed because it is too large Load Diff

3
context_encoding_model/_tp0_bk3/graph.neff Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:88e20e9aacad85427bc64b564935a053a2d089c921c32c96c0f2363c14b1f8fb
size 1004544

9514
context_encoding_model/_tp0_bk3/log-neuron-cc.txt Normal file
View File

File diff suppressed because it is too large Load Diff

3
context_encoding_model/_tp0_bk3/metaneff.pb Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:f4bfe18bacca09ec836a16458059ad676212bf8842fbb89645dbe0b21f90bea0
size 2454034

3
context_encoding_model/_tp0_bk3/model.MODULE_0984f4c19a044cc11c2a+1c024d2c.hlo_module.pb Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:64bf60998b259cee8443492d313f8dd44a9b30e8f7465266150157a423e9164c
size 2540428

3
context_encoding_model/_tp0_bk3/model.MODULE_0984f4c19a044cc11c2a+1c024d2c.neff Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:88e20e9aacad85427bc64b564935a053a2d089c921c32c96c0f2363c14b1f8fb
size 1004544

224
context_encoding_model/_tp0_bk3/neuron_config.json Normal file
View File

@@ -0,0 +1,224 @@
{
"_attn_implementation_autoset": false,
"_name_or_path": "/models/Qwen3-1.7B/",
"add_cross_attention": false,
"architectures": [
"Qwen3ForCausalLM"
],
"attention_bias": false,
"attention_dropout": 0.0,
"attribute_map": {},
"bad_words_ids": null,
"begin_suppress_tokens": null,
"bos_token_id": 151643,
"chunk_size_feed_forward": 0,
"cross_attention_hidden_size": null,
"decoder_start_token_id": null,
"diversity_penalty": 0.0,
"do_sample": false,
"early_stopping": false,
"encoder_no_repeat_ngram_size": 0,
"eos_token_id": 151645,
"exponential_decay_length_penalty": null,
"finetuning_task": null,
"forced_bos_token_id": null,
"forced_eos_token_id": null,
"fused_spec_config": null,
"head_dim": 128,
"hidden_act": "silu",
"hidden_size": 2048,
"id2label": {
"0": "LABEL_0",
"1": "LABEL_1"
},
"initializer_range": 0.02,
"intermediate_size": 6144,
"is_decoder": false,
"is_encoder_decoder": false,
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1
},
"length_penalty": 1.0,
"max_length": 20,
"max_position_embeddings": 40960,
"max_window_layers": 28,
"metadata": null,
"min_length": 0,
"model_type": "qwen3",
"neuron_config": {
"activation_quantization_type": null,
"allow_input_truncation": false,
"apply_seq_ids_mask": false,
"async_mode": false,
"attention_dp_degree": 1,
"attention_dtype": null,
"attn_block_cte_nki_kernel_enabled": false,
"attn_block_tkg_nki_kernel_cache_update": false,
"attn_block_tkg_nki_kernel_cascaded_attention": false,
"attn_block_tkg_nki_kernel_enabled": false,
"attn_cls": {
"__module__": "neuronx_distributed_inference.models.qwen3.modeling_qwen3",
"__name__": "NeuronQwen3Attention"
},
"attn_kernel_enabled": null,
"attn_tkg_builtin_kernel_enabled": false,
"attn_tkg_nki_kernel_enabled": false,
"batch_size": 1,
"bucket_n_active_tokens": true,
"buckets": [
1024
],
"cast_type": "config",
"cc_pipeline_tiling_factor": 2,
"chunked_prefill_config": null,
"context_encoding_buckets": [
1024
],
"cp_degree": 1,
"ctx_batch_size": 1,
"disable_kv_cache_tiling": false,
"draft_model_modules_to_not_convert": null,
"enable_bucketing": true,
"enable_cte_modular_flow": false,
"enable_eagle_draft_input_norm": false,
"enable_eagle_speculation": false,
"enable_fused_speculation": false,
"enable_long_context_mode": false,
"enable_output_completion_notifications": false,
"enable_spill_reload_dge": false,
"enable_token_tree": false,
"ep_degree": 1,
"expert_mlp_nki_kernel_enabled": null,
"flash_decoding_enabled": false,
"fused_qkv": false,
"fused_rmsnorm_skip_gamma": false,
"is_block_kv_layout": null,
"is_chunked_prefill": false,
"is_continuous_batching": true,
"is_eagle_draft": false,
"is_medusa": false,
"is_prefill_stage": true,
"is_prefix_caching": false,
"k_cache_transposed": false,
"kv_cache_batch_size": 4,
"kv_cache_padding_size": 0,
"kv_cache_quant": false,
"kv_cache_tiling": false,
"layer_boundary_markers": false,
"lm_head_pad": true,
"lm_head_pad_alignment_size": 1,
"local_ranks_size": 4,
"logical_nc_config": 2,
"lora_config": null,
"max_batch_size": 4,
"max_context_length": 2048,
"max_length": 2048,
"max_new_tokens": null,
"medusa_speculation_length": 0,
"medusa_tree": null,
"mlp_kernel_enabled": false,
"mlp_kernel_fuse_residual_add": false,
"modules_to_not_convert": null,
"moe_fused_nki_kernel_enabled": null,
"n_active_tokens": 2048,
"n_positions": 2048,
"num_medusa_heads": 0,
"on_cpu": false,
"on_device_sampling_config": {
"deterministic": false,
"do_sample": false,
"dynamic": true,
"global_topk": 256,
"on_device_sampling_config": true,
"temperature": 1.0,
"top_k": 1,
"top_k_kernel_enabled": false,
"top_p": 1.0
},
"output_logits": false,
"overrides_torch_dtype": true,
"pa_block_size": 2048,
"pa_num_blocks": 4,
"padding_side": "right",
"pp_degree": 1,
"prefix_buckets": null,
"qk_layernorm": false,
"qkv_kernel_enabled": false,
"qkv_kernel_fuse_residual_add": false,
"qkv_kernel_nbsd_layout": false,
"quantization_dtype": "int8",
"quantization_type": "per_tensor_symmetric",
"quantize_clamp_bound": Infinity,
"quantized": false,
"quantized_checkpoints_path": null,
"quantized_mlp_kernel_enabled": false,
"rmsnorm_quantize_kernel_enabled": false,
"router_topk_nki_kernel_enabled": null,
"rpl_reduce_dtype": null,
"save_sharded_checkpoint": true,
"scratchpad_page_size": null,
"seq_len": 2048,
"seq_len_threshold_for_cc_tiling": 16384,
"sequence_parallel_enabled": false,
"shared_mlp_nki_kernel_enabled": null,
"skip_sharding": false,
"skip_warmup": false,
"spec_batch_size": 4,
"speculation_length": 0,
"start_rank_id": 0,
"strided_context_parallel_kernel_enabled": false,
"target": null,
"tensor_capture_config": null,
"tile_cc": false,
"tkg_batch_size": 4,
"token_generation_buckets": null,
"token_tree_config": null,
"torch_dtype": "bfloat16",
"tp_degree": 4,
"vocab_parallel": false,
"weight_gather_seq_len_threshold": 32768,
"weights_to_skip_layout_optimization": [],
"world_size": 4
},
"no_repeat_ngram_size": 0,
"num_attention_heads": 16,
"num_beam_groups": 1,
"num_beams": 1,
"num_cores_per_group": 1,
"num_hidden_layers": 28,
"num_key_value_heads": 8,
"num_return_sequences": 1,
"output_attentions": false,
"output_hidden_states": false,
"output_scores": false,
"pad_token_id": 0,
"prefix": null,
"problem_type": null,
"pruned_heads": {},
"remove_invalid_values": false,
"repetition_penalty": 1.0,
"return_dict": true,
"return_dict_in_generate": false,
"rms_norm_eps": 1e-06,
"rope_scaling": null,
"rope_theta": 1000000,
"sep_token_id": null,
"sliding_window": null,
"suppress_tokens": null,
"task_specific_params": null,
"temperature": 1.0,
"tf_legacy_loss": false,
"tie_encoder_decoder": false,
"tie_word_embeddings": true,
"tokenizer_class": null,
"top_k": 50,
"top_p": 1.0,
"torchscript": false,
"transformers_version": "4.51.0",
"typical_p": 1.0,
"use_bfloat16": false,
"use_cache": true,
"use_sliding_window": false,
"vocab_size": 151936
}

1
context_encoding_model/_tp0_bk4/command.txt Normal file
View File

@@ -0,0 +1 @@
neuronx-cc compile --framework=XLA model.MODULE_e6b44ff520e6c4333666+e9aa1481.hlo_module.pb --output model.MODULE_e6b44ff520e6c4333666+e9aa1481.neff --target=trn2 --auto-cast=none --model-type=transformer '--tensorizer-options=--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-strided-dma ' --lnc=2 -O1 '--internal-hlo2tensorizer-options= --modular-flow-mac-threshold=10 --verify-hlo=true' --logfile=log-neuron-cc.txt --verbose=35

1
context_encoding_model/_tp0_bk4/compile_flags.MODULE_e6b44ff520e6c4333666+e9aa1481.json Normal file
View File

@@ -0,0 +1 @@
["--target=trn2", "--auto-cast=none", "--model-type=transformer", "--tensorizer-options=--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-strided-dma ", "--lnc=2", "-O1", "--internal-hlo2tensorizer-options= --modular-flow-mac-threshold=10 --verify-hlo=true", "--logfile=/models/Qwen3-1.7B-compiled/context_encoding_model/_tp0_bk4/log-neuron-cc.txt"]

1177
context_encoding_model/_tp0_bk4/global_metric_store.json Normal file
View File

File diff suppressed because it is too large Load Diff

3
context_encoding_model/_tp0_bk4/graph.neff Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:f1666f119851fd3de7f6ea4cb5da22169bcf4b1284acdd2f9f3afdc8e33da8b0
size 1250304

9501
context_encoding_model/_tp0_bk4/log-neuron-cc.txt Normal file
View File

File diff suppressed because it is too large Load Diff

3
context_encoding_model/_tp0_bk4/metaneff.pb Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:352da1073d15c0fbd9355ed2074a8ee0012540534b027fe615c424d66b1ee859
size 2798098

3
context_encoding_model/_tp0_bk4/model.MODULE_e6b44ff520e6c4333666+e9aa1481.hlo_module.pb Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:91f063d37769930b9be102cb6a84963b790e2ac12eb75d4e6018c2b479cf533d
size 2884492

3
context_encoding_model/_tp0_bk4/model.MODULE_e6b44ff520e6c4333666+e9aa1481.neff Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:f1666f119851fd3de7f6ea4cb5da22169bcf4b1284acdd2f9f3afdc8e33da8b0
size 1250304

224
context_encoding_model/_tp0_bk4/neuron_config.json Normal file
View File

@@ -0,0 +1,224 @@
{
"_attn_implementation_autoset": false,
"_name_or_path": "/models/Qwen3-1.7B/",
"add_cross_attention": false,
"architectures": [
"Qwen3ForCausalLM"
],
"attention_bias": false,
"attention_dropout": 0.0,
"attribute_map": {},
"bad_words_ids": null,
"begin_suppress_tokens": null,
"bos_token_id": 151643,
"chunk_size_feed_forward": 0,
"cross_attention_hidden_size": null,
"decoder_start_token_id": null,
"diversity_penalty": 0.0,
"do_sample": false,
"early_stopping": false,
"encoder_no_repeat_ngram_size": 0,
"eos_token_id": 151645,
"exponential_decay_length_penalty": null,
"finetuning_task": null,
"forced_bos_token_id": null,
"forced_eos_token_id": null,
"fused_spec_config": null,
"head_dim": 128,
"hidden_act": "silu",
"hidden_size": 2048,
"id2label": {
"0": "LABEL_0",
"1": "LABEL_1"
},
"initializer_range": 0.02,
"intermediate_size": 6144,
"is_decoder": false,
"is_encoder_decoder": false,
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1
},
"length_penalty": 1.0,
"max_length": 20,
"max_position_embeddings": 40960,
"max_window_layers": 28,
"metadata": null,
"min_length": 0,
"model_type": "qwen3",
"neuron_config": {
"activation_quantization_type": null,
"allow_input_truncation": false,
"apply_seq_ids_mask": false,
"async_mode": false,
"attention_dp_degree": 1,
"attention_dtype": null,
"attn_block_cte_nki_kernel_enabled": false,
"attn_block_tkg_nki_kernel_cache_update": false,
"attn_block_tkg_nki_kernel_cascaded_attention": false,
"attn_block_tkg_nki_kernel_enabled": false,
"attn_cls": {
"__module__": "neuronx_distributed_inference.models.qwen3.modeling_qwen3",
"__name__": "NeuronQwen3Attention"
},
"attn_kernel_enabled": null,
"attn_tkg_builtin_kernel_enabled": false,
"attn_tkg_nki_kernel_enabled": false,
"batch_size": 1,
"bucket_n_active_tokens": true,
"buckets": [
2048
],
"cast_type": "config",
"cc_pipeline_tiling_factor": 2,
"chunked_prefill_config": null,
"context_encoding_buckets": [
2048
],
"cp_degree": 1,
"ctx_batch_size": 1,
"disable_kv_cache_tiling": false,
"draft_model_modules_to_not_convert": null,
"enable_bucketing": true,
"enable_cte_modular_flow": false,
"enable_eagle_draft_input_norm": false,
"enable_eagle_speculation": false,
"enable_fused_speculation": false,
"enable_long_context_mode": false,
"enable_output_completion_notifications": false,
"enable_spill_reload_dge": false,
"enable_token_tree": false,
"ep_degree": 1,
"expert_mlp_nki_kernel_enabled": null,
"flash_decoding_enabled": false,
"fused_qkv": false,
"fused_rmsnorm_skip_gamma": false,
"is_block_kv_layout": null,
"is_chunked_prefill": false,
"is_continuous_batching": true,
"is_eagle_draft": false,
"is_medusa": false,
"is_prefill_stage": true,
"is_prefix_caching": false,
"k_cache_transposed": false,
"kv_cache_batch_size": 4,
"kv_cache_padding_size": 0,
"kv_cache_quant": false,
"kv_cache_tiling": false,
"layer_boundary_markers": false,
"lm_head_pad": true,
"lm_head_pad_alignment_size": 1,
"local_ranks_size": 4,
"logical_nc_config": 2,
"lora_config": null,
"max_batch_size": 4,
"max_context_length": 2048,
"max_length": 2048,
"max_new_tokens": null,
"medusa_speculation_length": 0,
"medusa_tree": null,
"mlp_kernel_enabled": false,
"mlp_kernel_fuse_residual_add": false,
"modules_to_not_convert": null,
"moe_fused_nki_kernel_enabled": null,
"n_active_tokens": 2048,
"n_positions": 2048,
"num_medusa_heads": 0,
"on_cpu": false,
"on_device_sampling_config": {
"deterministic": false,
"do_sample": false,
"dynamic": true,
"global_topk": 256,
"on_device_sampling_config": true,
"temperature": 1.0,
"top_k": 1,
"top_k_kernel_enabled": false,
"top_p": 1.0
},
"output_logits": false,
"overrides_torch_dtype": true,
"pa_block_size": 2048,
"pa_num_blocks": 4,
"padding_side": "right",
"pp_degree": 1,
"prefix_buckets": null,
"qk_layernorm": false,
"qkv_kernel_enabled": false,
"qkv_kernel_fuse_residual_add": false,
"qkv_kernel_nbsd_layout": false,
"quantization_dtype": "int8",
"quantization_type": "per_tensor_symmetric",
"quantize_clamp_bound": Infinity,
"quantized": false,
"quantized_checkpoints_path": null,
"quantized_mlp_kernel_enabled": false,
"rmsnorm_quantize_kernel_enabled": false,
"router_topk_nki_kernel_enabled": null,
"rpl_reduce_dtype": null,
"save_sharded_checkpoint": true,
"scratchpad_page_size": null,
"seq_len": 2048,
"seq_len_threshold_for_cc_tiling": 16384,
"sequence_parallel_enabled": false,
"shared_mlp_nki_kernel_enabled": null,
"skip_sharding": false,
"skip_warmup": false,
"spec_batch_size": 4,
"speculation_length": 0,
"start_rank_id": 0,
"strided_context_parallel_kernel_enabled": false,
"target": null,
"tensor_capture_config": null,
"tile_cc": false,
"tkg_batch_size": 4,
"token_generation_buckets": null,
"token_tree_config": null,
"torch_dtype": "bfloat16",
"tp_degree": 4,
"vocab_parallel": false,
"weight_gather_seq_len_threshold": 32768,
"weights_to_skip_layout_optimization": [],
"world_size": 4
},
"no_repeat_ngram_size": 0,
"num_attention_heads": 16,
"num_beam_groups": 1,
"num_beams": 1,
"num_cores_per_group": 1,
"num_hidden_layers": 28,
"num_key_value_heads": 8,
"num_return_sequences": 1,
"output_attentions": false,
"output_hidden_states": false,
"output_scores": false,
"pad_token_id": 0,
"prefix": null,
"problem_type": null,
"pruned_heads": {},
"remove_invalid_values": false,
"repetition_penalty": 1.0,
"return_dict": true,
"return_dict_in_generate": false,
"rms_norm_eps": 1e-06,
"rope_scaling": null,
"rope_theta": 1000000,
"sep_token_id": null,
"sliding_window": null,
"suppress_tokens": null,
"task_specific_params": null,
"temperature": 1.0,
"tf_legacy_loss": false,
"tie_encoder_decoder": false,
"tie_word_embeddings": true,
"tokenizer_class": null,
"top_k": 50,
"top_p": 1.0,
"torchscript": false,
"transformers_version": "4.51.0",
"typical_p": 1.0,
"use_bfloat16": false,
"use_cache": true,
"use_sliding_window": false,
"vocab_size": 151936
}