Synchronizing local compiler cache.
Browse files- .gitattributes +10 -0
- neuronxcc-2.21.33363.0+82129205/0_REGISTRY/0.4.6.dev4/179f57339300218ee5f1e16199abfeaede9aa891026ff77f3e8e13d9e193a48f/b4b9b98e1b50f733a7a8.json +97 -0
- neuronxcc-2.21.33363.0+82129205/0_REGISTRY/0.4.6.dev4/441269935591cad8d370e512c0b93cdd2fce6247c40e5a4866d872ee5338b0de/fc27c32a0c4ec3f887e9.json +82 -0
- neuronxcc-2.21.33363.0+82129205/0_REGISTRY/0.4.6.dev4/d2eb4ac153ba8d26f615a30665abec77e91fe75a5262767a548a93ec5fbcccbc/24596f911d11ff4166b8.json +97 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_3164b243bf918aa0b477+24129607/compile_flags.json +1 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_3164b243bf918aa0b477+24129607/model.done +0 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_3164b243bf918aa0b477+24129607/model.hlo_module.pb +3 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_3164b243bf918aa0b477+24129607/model.neff +3 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_4f8ed259d620f5f869d0+a02c3a36/compile_flags.json +1 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_4f8ed259d620f5f869d0+a02c3a36/model.done +0 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_4f8ed259d620f5f869d0+a02c3a36/model.hlo_module.pb +3 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_4f8ed259d620f5f869d0+a02c3a36/model.neff +3 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_4f8ed259d620f5f869d0+a02c3a36/wrapped_neff.hlo +3 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_5f5b72ab3e0b8d826f02+c4f887dc/compile_flags.json +1 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_5f5b72ab3e0b8d826f02+c4f887dc/model.done +0 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_5f5b72ab3e0b8d826f02+c4f887dc/model.hlo_module.pb +3 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_5f5b72ab3e0b8d826f02+c4f887dc/model.neff +3 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_5f5b72ab3e0b8d826f02+c4f887dc/wrapped_neff.hlo +3 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_78aec5e2f32f6fc831aa+a02c3a36/compile_flags.json +1 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_78aec5e2f32f6fc831aa+a02c3a36/model.done +0 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_78aec5e2f32f6fc831aa+a02c3a36/model.hlo_module.pb +3 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_78aec5e2f32f6fc831aa+a02c3a36/model.neff +3 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_78aec5e2f32f6fc831aa+a02c3a36/wrapped_neff.hlo +3 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_a02e663f9d5e1913e9e3+24129607/compile_flags.json +1 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_a02e663f9d5e1913e9e3+24129607/model.done +0 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_a02e663f9d5e1913e9e3+24129607/model.hlo_module.pb +3 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_a02e663f9d5e1913e9e3+24129607/model.neff +3 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_c480f8583bce4a388b93+c4f887dc/compile_flags.json +1 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_c480f8583bce4a388b93+c4f887dc/model.done +0 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_c480f8583bce4a388b93+c4f887dc/model.hlo_module.pb +3 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_c480f8583bce4a388b93+c4f887dc/model.neff +3 -0
- neuronxcc-2.21.33363.0+82129205/MODULE_c480f8583bce4a388b93+c4f887dc/wrapped_neff.hlo +3 -0
.gitattributes
CHANGED
|
@@ -7890,3 +7890,13 @@ neuronxcc-2.21.33363.0+82129205/MODULE_9b7eb59becdba58feaf4+c4f887dc/wrapped_nef
|
|
| 7890 |
neuronxcc-2.21.33363.0+82129205/MODULE_531086d2d1e6a2d23ebd+24129607/model.neff filter=lfs diff=lfs merge=lfs -text
|
| 7891 |
neuronxcc-2.21.33363.0+82129205/MODULE_d962f9089341da526ccb+a02c3a36/model.neff filter=lfs diff=lfs merge=lfs -text
|
| 7892 |
neuronxcc-2.21.33363.0+82129205/MODULE_d962f9089341da526ccb+a02c3a36/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7890 |
neuronxcc-2.21.33363.0+82129205/MODULE_531086d2d1e6a2d23ebd+24129607/model.neff filter=lfs diff=lfs merge=lfs -text
|
| 7891 |
neuronxcc-2.21.33363.0+82129205/MODULE_d962f9089341da526ccb+a02c3a36/model.neff filter=lfs diff=lfs merge=lfs -text
|
| 7892 |
neuronxcc-2.21.33363.0+82129205/MODULE_d962f9089341da526ccb+a02c3a36/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
| 7893 |
+
neuronxcc-2.21.33363.0+82129205/MODULE_3164b243bf918aa0b477+24129607/model.neff filter=lfs diff=lfs merge=lfs -text
|
| 7894 |
+
neuronxcc-2.21.33363.0+82129205/MODULE_4f8ed259d620f5f869d0+a02c3a36/model.neff filter=lfs diff=lfs merge=lfs -text
|
| 7895 |
+
neuronxcc-2.21.33363.0+82129205/MODULE_4f8ed259d620f5f869d0+a02c3a36/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
| 7896 |
+
neuronxcc-2.21.33363.0+82129205/MODULE_5f5b72ab3e0b8d826f02+c4f887dc/model.neff filter=lfs diff=lfs merge=lfs -text
|
| 7897 |
+
neuronxcc-2.21.33363.0+82129205/MODULE_5f5b72ab3e0b8d826f02+c4f887dc/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
| 7898 |
+
neuronxcc-2.21.33363.0+82129205/MODULE_78aec5e2f32f6fc831aa+a02c3a36/model.neff filter=lfs diff=lfs merge=lfs -text
|
| 7899 |
+
neuronxcc-2.21.33363.0+82129205/MODULE_78aec5e2f32f6fc831aa+a02c3a36/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
| 7900 |
+
neuronxcc-2.21.33363.0+82129205/MODULE_a02e663f9d5e1913e9e3+24129607/model.neff filter=lfs diff=lfs merge=lfs -text
|
| 7901 |
+
neuronxcc-2.21.33363.0+82129205/MODULE_c480f8583bce4a388b93+c4f887dc/model.neff filter=lfs diff=lfs merge=lfs -text
|
| 7902 |
+
neuronxcc-2.21.33363.0+82129205/MODULE_c480f8583bce4a388b93+c4f887dc/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
neuronxcc-2.21.33363.0+82129205/0_REGISTRY/0.4.6.dev4/179f57339300218ee5f1e16199abfeaede9aa891026ff77f3e8e13d9e193a48f/b4b9b98e1b50f733a7a8.json
ADDED
|
@@ -0,0 +1,97 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_entry_class": "SingleModelCacheEntry",
|
| 3 |
+
"_model_id": "tengomucho/tiny-random-gemma-3-vlm",
|
| 4 |
+
"_task": "image-text-to-text",
|
| 5 |
+
"architectures": [
|
| 6 |
+
"Gemma3ForConditionalGeneration"
|
| 7 |
+
],
|
| 8 |
+
"boi_token_index": 255999,
|
| 9 |
+
"dtype": "bfloat16",
|
| 10 |
+
"eoi_token_index": 256000,
|
| 11 |
+
"image_token_index": 262144,
|
| 12 |
+
"initializer_range": 0.02,
|
| 13 |
+
"mm_tokens_per_image": 256,
|
| 14 |
+
"model_type": "gemma3",
|
| 15 |
+
"neuron": {
|
| 16 |
+
"_serialized_key": "NxDVLMNeuronConfig",
|
| 17 |
+
"batch_size": 1,
|
| 18 |
+
"capacity_factor": null,
|
| 19 |
+
"checkpoint_id": "tengomucho/tiny-random-gemma-3-vlm",
|
| 20 |
+
"checkpoint_revision": "fbd846052676e5f7b417d7df6a736f5e980106b2",
|
| 21 |
+
"continuous_batching": false,
|
| 22 |
+
"ep_degree": 1,
|
| 23 |
+
"fused_qkv": true,
|
| 24 |
+
"glu_mlp": true,
|
| 25 |
+
"image_seq_len": 256,
|
| 26 |
+
"image_size": 896,
|
| 27 |
+
"local_ranks_size": 1,
|
| 28 |
+
"max_batch_size": 1,
|
| 29 |
+
"max_context_length": 2048,
|
| 30 |
+
"max_num_images": 5,
|
| 31 |
+
"max_topk": 256,
|
| 32 |
+
"n_active_tokens": 2048,
|
| 33 |
+
"neuronxcc_version": "2.21.33363.0+82129205",
|
| 34 |
+
"on_device_sampling": true,
|
| 35 |
+
"optimum_neuron_version": "0.4.6.dev4",
|
| 36 |
+
"output_logits": false,
|
| 37 |
+
"pp_degree": 1,
|
| 38 |
+
"prefill_chunk_size": 0,
|
| 39 |
+
"sequence_length": 2048,
|
| 40 |
+
"speculation_length": 0,
|
| 41 |
+
"start_rank_id": 0,
|
| 42 |
+
"target": "trn1",
|
| 43 |
+
"torch_dtype": "bfloat16",
|
| 44 |
+
"tp_degree": 1
|
| 45 |
+
},
|
| 46 |
+
"num_key_value_heads": 2,
|
| 47 |
+
"text_config": {
|
| 48 |
+
"_sliding_window_pattern": 2,
|
| 49 |
+
"attention_bias": false,
|
| 50 |
+
"attention_dropout": 0.0,
|
| 51 |
+
"attn_logit_softcapping": null,
|
| 52 |
+
"cache_implementation": "hybrid",
|
| 53 |
+
"dtype": "bfloat16",
|
| 54 |
+
"final_logit_softcapping": null,
|
| 55 |
+
"head_dim": 128,
|
| 56 |
+
"hidden_activation": "gelu_pytorch_tanh",
|
| 57 |
+
"hidden_size": 32,
|
| 58 |
+
"initializer_range": 0.02,
|
| 59 |
+
"intermediate_size": 128,
|
| 60 |
+
"layer_types": [
|
| 61 |
+
"sliding_attention",
|
| 62 |
+
"full_attention"
|
| 63 |
+
],
|
| 64 |
+
"max_position_embeddings": 131072,
|
| 65 |
+
"model_type": "gemma3_text",
|
| 66 |
+
"num_attention_heads": 4,
|
| 67 |
+
"num_hidden_layers": 2,
|
| 68 |
+
"num_key_value_heads": 1,
|
| 69 |
+
"query_pre_attn_scalar": 168,
|
| 70 |
+
"rms_norm_eps": 1e-06,
|
| 71 |
+
"rope_local_base_freq": 10000.0,
|
| 72 |
+
"rope_scaling": {
|
| 73 |
+
"factor": 8.0,
|
| 74 |
+
"rope_type": "linear"
|
| 75 |
+
},
|
| 76 |
+
"rope_theta": 1000000.0,
|
| 77 |
+
"sliding_window": 1024,
|
| 78 |
+
"sliding_window_pattern": 2,
|
| 79 |
+
"use_bidirectional_attention": false,
|
| 80 |
+
"use_cache": true,
|
| 81 |
+
"vocab_size": 262208
|
| 82 |
+
},
|
| 83 |
+
"vision_config": {
|
| 84 |
+
"attention_dropout": 0.0,
|
| 85 |
+
"hidden_act": "gelu_pytorch_tanh",
|
| 86 |
+
"hidden_size": 32,
|
| 87 |
+
"image_size": 896,
|
| 88 |
+
"intermediate_size": 128,
|
| 89 |
+
"layer_norm_eps": 1e-06,
|
| 90 |
+
"model_type": "siglip_vision_model",
|
| 91 |
+
"num_attention_heads": 2,
|
| 92 |
+
"num_channels": 3,
|
| 93 |
+
"num_hidden_layers": 2,
|
| 94 |
+
"patch_size": 14,
|
| 95 |
+
"vision_use_head": false
|
| 96 |
+
}
|
| 97 |
+
}
|
neuronxcc-2.21.33363.0+82129205/0_REGISTRY/0.4.6.dev4/441269935591cad8d370e512c0b93cdd2fce6247c40e5a4866d872ee5338b0de/fc27c32a0c4ec3f887e9.json
ADDED
|
@@ -0,0 +1,82 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_entry_class": "SingleModelCacheEntry",
|
| 3 |
+
"_model_id": "unsloth/gemma-3-270m-it",
|
| 4 |
+
"_sliding_window_pattern": 6,
|
| 5 |
+
"_task": "text-generation",
|
| 6 |
+
"architectures": [
|
| 7 |
+
"Gemma3ForCausalLM"
|
| 8 |
+
],
|
| 9 |
+
"attention_bias": false,
|
| 10 |
+
"attention_dropout": 0.0,
|
| 11 |
+
"attn_logit_softcapping": null,
|
| 12 |
+
"dtype": "bfloat16",
|
| 13 |
+
"final_logit_softcapping": null,
|
| 14 |
+
"head_dim": 256,
|
| 15 |
+
"hidden_activation": "gelu_pytorch_tanh",
|
| 16 |
+
"hidden_size": 640,
|
| 17 |
+
"initializer_range": 0.02,
|
| 18 |
+
"intermediate_size": 2048,
|
| 19 |
+
"layer_types": [
|
| 20 |
+
"sliding_attention",
|
| 21 |
+
"sliding_attention",
|
| 22 |
+
"sliding_attention",
|
| 23 |
+
"sliding_attention",
|
| 24 |
+
"sliding_attention",
|
| 25 |
+
"full_attention",
|
| 26 |
+
"sliding_attention",
|
| 27 |
+
"sliding_attention",
|
| 28 |
+
"sliding_attention",
|
| 29 |
+
"sliding_attention",
|
| 30 |
+
"sliding_attention",
|
| 31 |
+
"full_attention",
|
| 32 |
+
"sliding_attention",
|
| 33 |
+
"sliding_attention",
|
| 34 |
+
"sliding_attention",
|
| 35 |
+
"sliding_attention",
|
| 36 |
+
"sliding_attention",
|
| 37 |
+
"full_attention"
|
| 38 |
+
],
|
| 39 |
+
"max_position_embeddings": 32768,
|
| 40 |
+
"model_type": "gemma3_text",
|
| 41 |
+
"neuron": {
|
| 42 |
+
"_serialized_key": "NxDNeuronConfig",
|
| 43 |
+
"batch_size": 1,
|
| 44 |
+
"capacity_factor": null,
|
| 45 |
+
"checkpoint_id": "unsloth/gemma-3-270m-it",
|
| 46 |
+
"checkpoint_revision": "23cf460f6bb16954176b3ddcc8d4f250501458a9",
|
| 47 |
+
"continuous_batching": false,
|
| 48 |
+
"ep_degree": 1,
|
| 49 |
+
"fused_qkv": true,
|
| 50 |
+
"glu_mlp": true,
|
| 51 |
+
"local_ranks_size": 2,
|
| 52 |
+
"max_batch_size": 1,
|
| 53 |
+
"max_context_length": 512,
|
| 54 |
+
"max_topk": 256,
|
| 55 |
+
"n_active_tokens": 512,
|
| 56 |
+
"neuronxcc_version": "2.21.33363.0+82129205",
|
| 57 |
+
"on_device_sampling": true,
|
| 58 |
+
"optimum_neuron_version": "0.4.6.dev4",
|
| 59 |
+
"output_logits": false,
|
| 60 |
+
"pp_degree": 1,
|
| 61 |
+
"prefill_chunk_size": 0,
|
| 62 |
+
"sequence_length": 512,
|
| 63 |
+
"speculation_length": 0,
|
| 64 |
+
"start_rank_id": 0,
|
| 65 |
+
"target": "trn1",
|
| 66 |
+
"torch_dtype": "bfloat16",
|
| 67 |
+
"tp_degree": 2
|
| 68 |
+
},
|
| 69 |
+
"num_attention_heads": 4,
|
| 70 |
+
"num_hidden_layers": 18,
|
| 71 |
+
"num_key_value_heads": 1,
|
| 72 |
+
"query_pre_attn_scalar": 256,
|
| 73 |
+
"rms_norm_eps": 1e-06,
|
| 74 |
+
"rope_local_base_freq": 10000.0,
|
| 75 |
+
"rope_scaling": null,
|
| 76 |
+
"rope_theta": 1000000.0,
|
| 77 |
+
"sliding_window": 512,
|
| 78 |
+
"unsloth_fixed": true,
|
| 79 |
+
"use_bidirectional_attention": false,
|
| 80 |
+
"use_cache": true,
|
| 81 |
+
"vocab_size": 262144
|
| 82 |
+
}
|
neuronxcc-2.21.33363.0+82129205/0_REGISTRY/0.4.6.dev4/d2eb4ac153ba8d26f615a30665abec77e91fe75a5262767a548a93ec5fbcccbc/24596f911d11ff4166b8.json
ADDED
|
@@ -0,0 +1,97 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_entry_class": "SingleModelCacheEntry",
|
| 3 |
+
"_model_id": "tengomucho/tiny-random-gemma-3-vlm",
|
| 4 |
+
"_task": "image-text-to-text",
|
| 5 |
+
"architectures": [
|
| 6 |
+
"Gemma3ForConditionalGeneration"
|
| 7 |
+
],
|
| 8 |
+
"boi_token_index": 255999,
|
| 9 |
+
"dtype": "bfloat16",
|
| 10 |
+
"eoi_token_index": 256000,
|
| 11 |
+
"image_token_index": 262144,
|
| 12 |
+
"initializer_range": 0.02,
|
| 13 |
+
"mm_tokens_per_image": 256,
|
| 14 |
+
"model_type": "gemma3",
|
| 15 |
+
"neuron": {
|
| 16 |
+
"_serialized_key": "NxDVLMNeuronConfig",
|
| 17 |
+
"batch_size": 2,
|
| 18 |
+
"capacity_factor": null,
|
| 19 |
+
"checkpoint_id": "tengomucho/tiny-random-gemma-3-vlm",
|
| 20 |
+
"checkpoint_revision": "880db0f4a0e1000cec5193ea9a93c30ce425c0e5",
|
| 21 |
+
"continuous_batching": true,
|
| 22 |
+
"ep_degree": 1,
|
| 23 |
+
"fused_qkv": true,
|
| 24 |
+
"glu_mlp": true,
|
| 25 |
+
"image_seq_len": 256,
|
| 26 |
+
"image_size": 896,
|
| 27 |
+
"local_ranks_size": 2,
|
| 28 |
+
"max_batch_size": 2,
|
| 29 |
+
"max_context_length": 2048,
|
| 30 |
+
"max_num_images": 5,
|
| 31 |
+
"max_topk": 256,
|
| 32 |
+
"n_active_tokens": 2048,
|
| 33 |
+
"neuronxcc_version": "2.21.33363.0+82129205",
|
| 34 |
+
"on_device_sampling": true,
|
| 35 |
+
"optimum_neuron_version": "0.4.6.dev4",
|
| 36 |
+
"output_logits": false,
|
| 37 |
+
"pp_degree": 1,
|
| 38 |
+
"prefill_chunk_size": 0,
|
| 39 |
+
"sequence_length": 2048,
|
| 40 |
+
"speculation_length": 0,
|
| 41 |
+
"start_rank_id": 0,
|
| 42 |
+
"target": "trn1",
|
| 43 |
+
"torch_dtype": "bfloat16",
|
| 44 |
+
"tp_degree": 2
|
| 45 |
+
},
|
| 46 |
+
"num_key_value_heads": 2,
|
| 47 |
+
"text_config": {
|
| 48 |
+
"_sliding_window_pattern": 2,
|
| 49 |
+
"attention_bias": false,
|
| 50 |
+
"attention_dropout": 0.0,
|
| 51 |
+
"attn_logit_softcapping": null,
|
| 52 |
+
"cache_implementation": "hybrid",
|
| 53 |
+
"dtype": "bfloat16",
|
| 54 |
+
"final_logit_softcapping": null,
|
| 55 |
+
"head_dim": 64,
|
| 56 |
+
"hidden_activation": "gelu_pytorch_tanh",
|
| 57 |
+
"hidden_size": 128,
|
| 58 |
+
"initializer_range": 0.02,
|
| 59 |
+
"intermediate_size": 256,
|
| 60 |
+
"layer_types": [
|
| 61 |
+
"sliding_attention",
|
| 62 |
+
"full_attention"
|
| 63 |
+
],
|
| 64 |
+
"max_position_embeddings": 131072,
|
| 65 |
+
"model_type": "gemma3_text",
|
| 66 |
+
"num_attention_heads": 4,
|
| 67 |
+
"num_hidden_layers": 2,
|
| 68 |
+
"num_key_value_heads": 1,
|
| 69 |
+
"query_pre_attn_scalar": 168,
|
| 70 |
+
"rms_norm_eps": 1e-06,
|
| 71 |
+
"rope_local_base_freq": 10000.0,
|
| 72 |
+
"rope_scaling": {
|
| 73 |
+
"factor": 8.0,
|
| 74 |
+
"rope_type": "linear"
|
| 75 |
+
},
|
| 76 |
+
"rope_theta": 1000000.0,
|
| 77 |
+
"sliding_window": 1024,
|
| 78 |
+
"sliding_window_pattern": 2,
|
| 79 |
+
"use_bidirectional_attention": false,
|
| 80 |
+
"use_cache": true,
|
| 81 |
+
"vocab_size": 262208
|
| 82 |
+
},
|
| 83 |
+
"vision_config": {
|
| 84 |
+
"attention_dropout": 0.0,
|
| 85 |
+
"hidden_act": "gelu_pytorch_tanh",
|
| 86 |
+
"hidden_size": 32,
|
| 87 |
+
"image_size": 896,
|
| 88 |
+
"intermediate_size": 128,
|
| 89 |
+
"layer_norm_eps": 1e-06,
|
| 90 |
+
"model_type": "siglip_vision_model",
|
| 91 |
+
"num_attention_heads": 2,
|
| 92 |
+
"num_channels": 3,
|
| 93 |
+
"num_hidden_layers": 2,
|
| 94 |
+
"patch_size": 14,
|
| 95 |
+
"vision_use_head": false
|
| 96 |
+
}
|
| 97 |
+
}
|
neuronxcc-2.21.33363.0+82129205/MODULE_3164b243bf918aa0b477+24129607/compile_flags.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
["--target=trn1", "--auto-cast=none", "--model-type=transformer", "--tensorizer-options=--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-strided-dma ", "-O2", "--lnc=1", "--logfile=/tmp/nxd_model/context_encoding/_tp0_bk0/log-neuron-cc.txt"]
|
neuronxcc-2.21.33363.0+82129205/MODULE_3164b243bf918aa0b477+24129607/model.done
ADDED
|
File without changes
|
neuronxcc-2.21.33363.0+82129205/MODULE_3164b243bf918aa0b477+24129607/model.hlo_module.pb
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:845b3ea51d1ffbcaf1dae071c83c4fdf7afebee0fc572f41b4ab461bb56eb283
|
| 3 |
+
size 96227
|
neuronxcc-2.21.33363.0+82129205/MODULE_3164b243bf918aa0b477+24129607/model.neff
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:aa1e4393e72c40199b02ef0700e6826a80f99aeb8d55a54f4ebb67e62c9fc0c0
|
| 3 |
+
size 1301504
|
neuronxcc-2.21.33363.0+82129205/MODULE_4f8ed259d620f5f869d0+a02c3a36/compile_flags.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
["--target=trn1", "--auto-cast=none", "--model-type=transformer", "--tensorizer-options=--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-strided-dma ", "-O2", "--lnc=1", "--logfile=/tmp/nxd_model/token_generation/_tp0_bk0/log-neuron-cc.txt", "--enable-internal-neff-wrapper"]
|
neuronxcc-2.21.33363.0+82129205/MODULE_4f8ed259d620f5f869d0+a02c3a36/model.done
ADDED
|
File without changes
|
neuronxcc-2.21.33363.0+82129205/MODULE_4f8ed259d620f5f869d0+a02c3a36/model.hlo_module.pb
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:48f798909e3233225caab6080b72e5f1e64b3699001301f91f04d24e0a39017e
|
| 3 |
+
size 110053
|
neuronxcc-2.21.33363.0+82129205/MODULE_4f8ed259d620f5f869d0+a02c3a36/model.neff
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:04b76bfbbf72f5e91caf122c5bb483b6f25393b4548acba385b2941f77ab8c3c
|
| 3 |
+
size 492544
|
neuronxcc-2.21.33363.0+82129205/MODULE_4f8ed259d620f5f869d0+a02c3a36/wrapped_neff.hlo
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c2a16f4f4ad9fe09e56abfea938eb27ae58676dc5feab143d720e3553c7bc29e
|
| 3 |
+
size 501517
|
neuronxcc-2.21.33363.0+82129205/MODULE_5f5b72ab3e0b8d826f02+c4f887dc/compile_flags.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
["--target=trn1", "--auto-cast=none", "--model-type=transformer", "--tensorizer-options=--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-strided-dma ", "-O2", "--lnc=1", "--logfile=/tmp/nxd_model/vision_encoder/_tp0_bk0/log-neuron-cc.txt", "--enable-internal-neff-wrapper"]
|
neuronxcc-2.21.33363.0+82129205/MODULE_5f5b72ab3e0b8d826f02+c4f887dc/model.done
ADDED
|
File without changes
|
neuronxcc-2.21.33363.0+82129205/MODULE_5f5b72ab3e0b8d826f02+c4f887dc/model.hlo_module.pb
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a064c6f2fc0df263d71479ccdeed7f50465e9f14030ec5334b9ac7bedd5e4420
|
| 3 |
+
size 42200
|
neuronxcc-2.21.33363.0+82129205/MODULE_5f5b72ab3e0b8d826f02+c4f887dc/model.neff
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f1565e9e1b0e1581535ad1df52c665d284f95291541b1e88ca49711b348fee7e
|
| 3 |
+
size 4404224
|
neuronxcc-2.21.33363.0+82129205/MODULE_5f5b72ab3e0b8d826f02+c4f887dc/wrapped_neff.hlo
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c5cf6e5611bb5d009acfb6cb6a6467d2f73bb51ffa65c7d2aa44bf863930098f
|
| 3 |
+
size 4415081
|
neuronxcc-2.21.33363.0+82129205/MODULE_78aec5e2f32f6fc831aa+a02c3a36/compile_flags.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
["--target=trn1", "--auto-cast=none", "--model-type=transformer", "--tensorizer-options=--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-strided-dma ", "-O2", "--lnc=1", "--logfile=/tmp/nxd_model/token_generation/_tp0_bk0/log-neuron-cc.txt", "--enable-internal-neff-wrapper"]
|
neuronxcc-2.21.33363.0+82129205/MODULE_78aec5e2f32f6fc831aa+a02c3a36/model.done
ADDED
|
File without changes
|
neuronxcc-2.21.33363.0+82129205/MODULE_78aec5e2f32f6fc831aa+a02c3a36/model.hlo_module.pb
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e899fd2bfa8bc130c686d925d195ccd3ec66d32116f3f4b9611b211c6d94444b
|
| 3 |
+
size 102577
|
neuronxcc-2.21.33363.0+82129205/MODULE_78aec5e2f32f6fc831aa+a02c3a36/model.neff
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:651e66023989ecce31f41745714e2fe351dcc074d55b67171c4a8665ac9b5f66
|
| 3 |
+
size 881664
|
neuronxcc-2.21.33363.0+82129205/MODULE_78aec5e2f32f6fc831aa+a02c3a36/wrapped_neff.hlo
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:76c6193ec8564aa7a37dea47e9a05b582e35db1e0f68ba59d34cf3fd0c9f08c2
|
| 3 |
+
size 890539
|
neuronxcc-2.21.33363.0+82129205/MODULE_a02e663f9d5e1913e9e3+24129607/compile_flags.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
["--target=trn1", "--auto-cast=none", "--model-type=transformer", "--tensorizer-options=--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-strided-dma ", "-O2", "--lnc=1", "--logfile=/tmp/nxd_model/context_encoding/_tp0_bk0/log-neuron-cc.txt"]
|
neuronxcc-2.21.33363.0+82129205/MODULE_a02e663f9d5e1913e9e3+24129607/model.done
ADDED
|
File without changes
|
neuronxcc-2.21.33363.0+82129205/MODULE_a02e663f9d5e1913e9e3+24129607/model.hlo_module.pb
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0ae29baeb064ec697a8a958f908b31b98f8e2f91d002827bd6809acd9b1c4fb1
|
| 3 |
+
size 107706
|
neuronxcc-2.21.33363.0+82129205/MODULE_a02e663f9d5e1913e9e3+24129607/model.neff
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1f6202fa09b0c67ee52dc345cd604e3d64666a98037f0026c541a84f2d0b49e6
|
| 3 |
+
size 656384
|
neuronxcc-2.21.33363.0+82129205/MODULE_c480f8583bce4a388b93+c4f887dc/compile_flags.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
["--target=trn1", "--auto-cast=none", "--model-type=transformer", "--tensorizer-options=--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-strided-dma ", "-O2", "--lnc=1", "--logfile=/tmp/nxd_model/vision_encoder/_tp0_bk0/log-neuron-cc.txt", "--enable-internal-neff-wrapper"]
|
neuronxcc-2.21.33363.0+82129205/MODULE_c480f8583bce4a388b93+c4f887dc/model.done
ADDED
|
File without changes
|
neuronxcc-2.21.33363.0+82129205/MODULE_c480f8583bce4a388b93+c4f887dc/model.hlo_module.pb
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d1136862d93c97e0882587ce7501a718e7e94c9abbab0cedcd11b8d7034b6f10
|
| 3 |
+
size 39717
|
neuronxcc-2.21.33363.0+82129205/MODULE_c480f8583bce4a388b93+c4f887dc/model.neff
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7701eb6ab62bfb715f6463985d5f4ee136cd9e335f285bdabef67aa131d5143f
|
| 3 |
+
size 2980864
|
neuronxcc-2.21.33363.0+82129205/MODULE_c480f8583bce4a388b93+c4f887dc/wrapped_neff.hlo
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:40265d84a046407b34b8f0513c9657bbe89f92cf11e8b34dd7aac05a53507e89
|
| 3 |
+
size 2993386
|