Synchronizing local compiler cache.
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- .gitattributes +51 -0
- neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/deepseek-ai/DeepSeek-R1-Distill-Llama-70B/16eb66e6f195b2f2f3b4.json +77 -0
- neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/deepseek-ai/DeepSeek-R1-Distill-Llama-70B/88744e5aa5d753d1f538.json +77 -0
- neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/deepseek-ai/DeepSeek-R1-Distill-Llama-70B/8a3e1095e7ec1ddfb65b.json +77 -0
- neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/llamafactory/tiny-random-Llama-3/021f11fdefa8c3b516bd.json +77 -0
- neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/llamafactory/tiny-random-Llama-3/1913cf3041a0fe975f3c.json +77 -0
- neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/llamafactory/tiny-random-Llama-3/50411def20a2b703209e.json +77 -0
- neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/llamafactory/tiny-random-Llama-3/7bca5f2b6f1034c1fa71.json +77 -0
- neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/meta-llama/Llama-3.2-1B-Instruct/34beaf9835b09fe53395.json +77 -0
- neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/meta-llama/Llama-3.2-1B-Instruct/d1ea2689244397f649b7.json +77 -0
- neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/unsloth/Llama-3.1-8B-Instruct/321051129e499a9d100e.json +78 -0
- neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/unsloth/Llama-3.1-8B-Instruct/59e56e036f276aac27ec.json +78 -0
- neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/unsloth/Llama-3.1-8B-Instruct/aed3ac4481c88779a26c.json +78 -0
- neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/unsloth/Llama-3.1-8B-Instruct/bcd80b4d12e05bc045ce.json +78 -0
- neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/unsloth/Llama-3.1-8B-Instruct/c28ab0c7d33e28708b3c.json +78 -0
- neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/unsloth/Llama-3.1-8B-Instruct/f0e18f873ce42aaa0b9d.json +78 -0
- neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/unsloth/Llama-3.1-8B-Instruct/fb1938af2d9e7e083207.json +78 -0
- neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev8/llama/meta-llama/Llama-3.2-1B-Instruct/3da196f521260e769ca6.json +77 -0
- neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev8/llama/unsloth/Llama-3.2-1B-Instruct/f4d3009ec7e739e5ca73.json +78 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_056a0c30057367f9353e+165e9558/compile_flags.json +1 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_056a0c30057367f9353e+165e9558/model.done +0 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_056a0c30057367f9353e+165e9558/model.hlo_module.pb +3 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_056a0c30057367f9353e+165e9558/model.neff +3 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_056a0c30057367f9353e+165e9558/wrapped_neff.hlo +3 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_07e4a454b533bd5b9f6f+bfe5714b/compile_flags.json +1 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_07e4a454b533bd5b9f6f+bfe5714b/model.done +0 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_07e4a454b533bd5b9f6f+bfe5714b/model.hlo_module.pb +3 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_07e4a454b533bd5b9f6f+bfe5714b/model.neff +3 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_08eb5012506859e6b129+165e9558/compile_flags.json +1 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_08eb5012506859e6b129+165e9558/model.done +0 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_08eb5012506859e6b129+165e9558/model.hlo_module.pb +3 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_08eb5012506859e6b129+165e9558/model.neff +3 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_08eb5012506859e6b129+165e9558/wrapped_neff.hlo +3 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_184d4ed11977011ce5e7+431f5505/compile_flags.json +1 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_184d4ed11977011ce5e7+431f5505/model.done +0 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_184d4ed11977011ce5e7+431f5505/model.hlo_module.pb +3 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_184d4ed11977011ce5e7+431f5505/model.neff +3 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_24ff9ac2787ce9a1d276+613edded/model.neff +1 -1
- neuronxcc-2.17.194.0+d312836f/MODULE_320f2622d4d0c9fdd0f1+613edded/model.neff +1 -1
- neuronxcc-2.17.194.0+d312836f/MODULE_374002f5e545876dbed9+bfe5714b/compile_flags.json +1 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_374002f5e545876dbed9+bfe5714b/model.done +0 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_374002f5e545876dbed9+bfe5714b/model.hlo_module.pb +3 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_374002f5e545876dbed9+bfe5714b/model.neff +3 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_40a0e75a65ac51fdd01a+613edded/model.neff +1 -1
- neuronxcc-2.17.194.0+d312836f/MODULE_4617ff1c720a11b46bd3+165e9558/compile_flags.json +1 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_4617ff1c720a11b46bd3+165e9558/model.done +0 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_4617ff1c720a11b46bd3+165e9558/model.hlo_module.pb +3 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_4617ff1c720a11b46bd3+165e9558/model.neff +3 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_4617ff1c720a11b46bd3+165e9558/wrapped_neff.hlo +3 -0
- neuronxcc-2.17.194.0+d312836f/MODULE_48bb18150a194c32e820+bfe5714b/compile_flags.json +1 -0
.gitattributes
CHANGED
@@ -2809,3 +2809,54 @@ neuronxcc-2.18.121.0+9e31e41a/MODULE_e3e99dc2a67feb5066f9+84f3e719/model.neff fi
|
|
2809 |
neuronxcc-2.17.194.0+d312836f/MODULE_306e43064416feb5de44+165e9558/model.neff filter=lfs diff=lfs merge=lfs -text
|
2810 |
neuronxcc-2.17.194.0+d312836f/MODULE_306e43064416feb5de44+165e9558/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
2811 |
neuronxcc-2.17.194.0+d312836f/MODULE_3d29f316a0cedf338ff9+bfe5714b/model.neff filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2809 |
neuronxcc-2.17.194.0+d312836f/MODULE_306e43064416feb5de44+165e9558/model.neff filter=lfs diff=lfs merge=lfs -text
|
2810 |
neuronxcc-2.17.194.0+d312836f/MODULE_306e43064416feb5de44+165e9558/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
2811 |
neuronxcc-2.17.194.0+d312836f/MODULE_3d29f316a0cedf338ff9+bfe5714b/model.neff filter=lfs diff=lfs merge=lfs -text
|
2812 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_056a0c30057367f9353e+165e9558/model.neff filter=lfs diff=lfs merge=lfs -text
|
2813 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_056a0c30057367f9353e+165e9558/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
2814 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_07e4a454b533bd5b9f6f+bfe5714b/model.neff filter=lfs diff=lfs merge=lfs -text
|
2815 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_08eb5012506859e6b129+165e9558/model.neff filter=lfs diff=lfs merge=lfs -text
|
2816 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_08eb5012506859e6b129+165e9558/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
2817 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_184d4ed11977011ce5e7+431f5505/model.neff filter=lfs diff=lfs merge=lfs -text
|
2818 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_374002f5e545876dbed9+bfe5714b/model.neff filter=lfs diff=lfs merge=lfs -text
|
2819 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_4617ff1c720a11b46bd3+165e9558/model.neff filter=lfs diff=lfs merge=lfs -text
|
2820 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_4617ff1c720a11b46bd3+165e9558/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
2821 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_48bb18150a194c32e820+bfe5714b/model.neff filter=lfs diff=lfs merge=lfs -text
|
2822 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_4d8202c650b98a0b9d0f+431f5505/model.neff filter=lfs diff=lfs merge=lfs -text
|
2823 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_5d6569b6be8918b2148b+bfe5714b/model.neff filter=lfs diff=lfs merge=lfs -text
|
2824 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_669cd724b366299c9441+165e9558/model.neff filter=lfs diff=lfs merge=lfs -text
|
2825 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_669cd724b366299c9441+165e9558/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
2826 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_66fce16545f84447c1f9+165e9558/model.neff filter=lfs diff=lfs merge=lfs -text
|
2827 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_66fce16545f84447c1f9+165e9558/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
2828 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_682b390c1c73a9a3bd11+165e9558/model.neff filter=lfs diff=lfs merge=lfs -text
|
2829 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_682b390c1c73a9a3bd11+165e9558/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
2830 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_75d614ebd3722304448a+431f5505/model.neff filter=lfs diff=lfs merge=lfs -text
|
2831 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_7b9581483dd1ab45e802+165e9558/model.neff filter=lfs diff=lfs merge=lfs -text
|
2832 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_7b9581483dd1ab45e802+165e9558/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
2833 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_7d5e63db6db4994c4291+bfe5714b/model.neff filter=lfs diff=lfs merge=lfs -text
|
2834 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_818dd5b285ecc79f51b6+165e9558/model.neff filter=lfs diff=lfs merge=lfs -text
|
2835 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_818dd5b285ecc79f51b6+165e9558/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
2836 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_88825b33884e22bf2350+165e9558/model.neff filter=lfs diff=lfs merge=lfs -text
|
2837 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_88825b33884e22bf2350+165e9558/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
2838 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_88c0c20acb62b9015ac5+bfe5714b/model.neff filter=lfs diff=lfs merge=lfs -text
|
2839 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_8bb6b9b7fb63c8407d1b+bfe5714b/model.neff filter=lfs diff=lfs merge=lfs -text
|
2840 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_94a6e940063af75dc1b5+bfe5714b/model.neff filter=lfs diff=lfs merge=lfs -text
|
2841 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_9e7d5afcefd712aef192+165e9558/model.neff filter=lfs diff=lfs merge=lfs -text
|
2842 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_9e7d5afcefd712aef192+165e9558/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
2843 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_a05cbb6e9bfee46f6408+431f5505/model.neff filter=lfs diff=lfs merge=lfs -text
|
2844 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_a446aa72fb4ce75a9fdf+bfe5714b/model.neff filter=lfs diff=lfs merge=lfs -text
|
2845 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_a6b9d38a6340a9ba7207+bfe5714b/model.neff filter=lfs diff=lfs merge=lfs -text
|
2846 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_c0a68b4d4ba9281a9229+bfe5714b/model.neff filter=lfs diff=lfs merge=lfs -text
|
2847 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_c0e52e55b5c34e6b0178+165e9558/model.neff filter=lfs diff=lfs merge=lfs -text
|
2848 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_c0e52e55b5c34e6b0178+165e9558/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
2849 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_c35c27d074e574f52183+165e9558/model.neff filter=lfs diff=lfs merge=lfs -text
|
2850 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_c35c27d074e574f52183+165e9558/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
2851 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_c5c5cbb467631d860a15+165e9558/model.neff filter=lfs diff=lfs merge=lfs -text
|
2852 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_c5c5cbb467631d860a15+165e9558/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
2853 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_cbb69a706480698b9414+165e9558/model.neff filter=lfs diff=lfs merge=lfs -text
|
2854 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_cbb69a706480698b9414+165e9558/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
2855 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_d0af0d798d24f975f633+bfe5714b/model.neff filter=lfs diff=lfs merge=lfs -text
|
2856 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_db2676d850a0b0c3f502+bfe5714b/model.neff filter=lfs diff=lfs merge=lfs -text
|
2857 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_e4125c11885b90ff94c9+431f5505/model.neff filter=lfs diff=lfs merge=lfs -text
|
2858 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_eb6d7bd59bec7840dd72+bfe5714b/model.neff filter=lfs diff=lfs merge=lfs -text
|
2859 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_f1f84223fca9c333bb2f+431f5505/model.neff filter=lfs diff=lfs merge=lfs -text
|
2860 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_f37bc22c32c557b09691+165e9558/model.neff filter=lfs diff=lfs merge=lfs -text
|
2861 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_f37bc22c32c557b09691+165e9558/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
|
2862 |
+
neuronxcc-2.17.194.0+d312836f/MODULE_f8a0bbe4c09b370fbea0+bfe5714b/model.neff filter=lfs diff=lfs merge=lfs -text
|
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/deepseek-ai/DeepSeek-R1-Distill-Llama-70B/16eb66e6f195b2f2f3b4.json
ADDED
@@ -0,0 +1,77 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_entry_class": "SingleModelCacheEntry",
|
3 |
+
"_model_id": "deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
|
4 |
+
"_task": "text-generation",
|
5 |
+
"architectures": [
|
6 |
+
"LlamaForCausalLM"
|
7 |
+
],
|
8 |
+
"attention_bias": false,
|
9 |
+
"attention_dropout": 0.0,
|
10 |
+
"head_dim": 128,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 8192,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 28672,
|
15 |
+
"max_position_embeddings": 131072,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"neuron": {
|
19 |
+
"_serialized_key": "NxDNeuronConfig",
|
20 |
+
"async_mode": false,
|
21 |
+
"attn_kernel_enabled": false,
|
22 |
+
"batch_size": 1,
|
23 |
+
"capacity_factor": null,
|
24 |
+
"cc_pipeline_tiling_factor": 2,
|
25 |
+
"checkpoint_id": "deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
|
26 |
+
"checkpoint_revision": "b1c0b44b4369b597ad119a196caf79a9c40e141e",
|
27 |
+
"continuous_batching": false,
|
28 |
+
"enable_bucketing": false,
|
29 |
+
"ep_degree": 1,
|
30 |
+
"flash_decoding_enabled": false,
|
31 |
+
"fused_qkv": true,
|
32 |
+
"glu_mlp": true,
|
33 |
+
"is_chunked_prefill": false,
|
34 |
+
"local_ranks_size": 24,
|
35 |
+
"logical_nc_config": 1,
|
36 |
+
"max_batch_size": 1,
|
37 |
+
"max_context_length": 4096,
|
38 |
+
"max_topk": 256,
|
39 |
+
"mlp_kernel_enabled": false,
|
40 |
+
"mlp_kernel_fuse_residual_add": false,
|
41 |
+
"n_active_tokens": 4096,
|
42 |
+
"neuronxcc_version": "2.17.194.0+d312836f",
|
43 |
+
"num_cores_per_group": 1,
|
44 |
+
"on_device_sampling": true,
|
45 |
+
"optimum_neuron_version": "0.2.0.dev7",
|
46 |
+
"output_logits": false,
|
47 |
+
"padding_side": "right",
|
48 |
+
"pp_degree": 1,
|
49 |
+
"qk_layernorm": false,
|
50 |
+
"qkv_kernel_enabled": false,
|
51 |
+
"rpl_reduce_dtype": "bfloat16",
|
52 |
+
"sequence_length": 4096,
|
53 |
+
"sequence_parallel_enabled": false,
|
54 |
+
"speculation_length": 0,
|
55 |
+
"start_rank_id": 0,
|
56 |
+
"target": null,
|
57 |
+
"torch_dtype": "bfloat16",
|
58 |
+
"tp_degree": 24,
|
59 |
+
"vocab_parallel": false
|
60 |
+
},
|
61 |
+
"num_attention_heads": 64,
|
62 |
+
"num_hidden_layers": 80,
|
63 |
+
"num_key_value_heads": 8,
|
64 |
+
"pretraining_tp": 1,
|
65 |
+
"rms_norm_eps": 1e-05,
|
66 |
+
"rope_scaling": {
|
67 |
+
"factor": 8.0,
|
68 |
+
"high_freq_factor": 4.0,
|
69 |
+
"low_freq_factor": 1.0,
|
70 |
+
"original_max_position_embeddings": 8192,
|
71 |
+
"rope_type": "llama3"
|
72 |
+
},
|
73 |
+
"rope_theta": 500000.0,
|
74 |
+
"tie_word_embeddings": false,
|
75 |
+
"use_cache": true,
|
76 |
+
"vocab_size": 128256
|
77 |
+
}
|
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/deepseek-ai/DeepSeek-R1-Distill-Llama-70B/88744e5aa5d753d1f538.json
ADDED
@@ -0,0 +1,77 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_entry_class": "SingleModelCacheEntry",
|
3 |
+
"_model_id": "deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
|
4 |
+
"_task": "text-generation",
|
5 |
+
"architectures": [
|
6 |
+
"LlamaForCausalLM"
|
7 |
+
],
|
8 |
+
"attention_bias": false,
|
9 |
+
"attention_dropout": 0.0,
|
10 |
+
"head_dim": 128,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 8192,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 28672,
|
15 |
+
"max_position_embeddings": 131072,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"neuron": {
|
19 |
+
"_serialized_key": "NxDNeuronConfig",
|
20 |
+
"async_mode": false,
|
21 |
+
"attn_kernel_enabled": false,
|
22 |
+
"batch_size": 4,
|
23 |
+
"capacity_factor": null,
|
24 |
+
"cc_pipeline_tiling_factor": 2,
|
25 |
+
"checkpoint_id": "deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
|
26 |
+
"checkpoint_revision": "b1c0b44b4369b597ad119a196caf79a9c40e141e",
|
27 |
+
"continuous_batching": true,
|
28 |
+
"enable_bucketing": false,
|
29 |
+
"ep_degree": 1,
|
30 |
+
"flash_decoding_enabled": false,
|
31 |
+
"fused_qkv": true,
|
32 |
+
"glu_mlp": true,
|
33 |
+
"is_chunked_prefill": false,
|
34 |
+
"local_ranks_size": 24,
|
35 |
+
"logical_nc_config": 1,
|
36 |
+
"max_batch_size": 4,
|
37 |
+
"max_context_length": 4096,
|
38 |
+
"max_topk": 256,
|
39 |
+
"mlp_kernel_enabled": false,
|
40 |
+
"mlp_kernel_fuse_residual_add": false,
|
41 |
+
"n_active_tokens": 4096,
|
42 |
+
"neuronxcc_version": "2.17.194.0+d312836f",
|
43 |
+
"num_cores_per_group": 1,
|
44 |
+
"on_device_sampling": true,
|
45 |
+
"optimum_neuron_version": "0.2.0.dev7",
|
46 |
+
"output_logits": false,
|
47 |
+
"padding_side": "right",
|
48 |
+
"pp_degree": 1,
|
49 |
+
"qk_layernorm": false,
|
50 |
+
"qkv_kernel_enabled": false,
|
51 |
+
"rpl_reduce_dtype": "bfloat16",
|
52 |
+
"sequence_length": 4096,
|
53 |
+
"sequence_parallel_enabled": false,
|
54 |
+
"speculation_length": 0,
|
55 |
+
"start_rank_id": 0,
|
56 |
+
"target": null,
|
57 |
+
"torch_dtype": "bfloat16",
|
58 |
+
"tp_degree": 24,
|
59 |
+
"vocab_parallel": false
|
60 |
+
},
|
61 |
+
"num_attention_heads": 64,
|
62 |
+
"num_hidden_layers": 80,
|
63 |
+
"num_key_value_heads": 8,
|
64 |
+
"pretraining_tp": 1,
|
65 |
+
"rms_norm_eps": 1e-05,
|
66 |
+
"rope_scaling": {
|
67 |
+
"factor": 8.0,
|
68 |
+
"high_freq_factor": 4.0,
|
69 |
+
"low_freq_factor": 1.0,
|
70 |
+
"original_max_position_embeddings": 8192,
|
71 |
+
"rope_type": "llama3"
|
72 |
+
},
|
73 |
+
"rope_theta": 500000.0,
|
74 |
+
"tie_word_embeddings": false,
|
75 |
+
"use_cache": true,
|
76 |
+
"vocab_size": 128256
|
77 |
+
}
|
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/deepseek-ai/DeepSeek-R1-Distill-Llama-70B/8a3e1095e7ec1ddfb65b.json
ADDED
@@ -0,0 +1,77 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_entry_class": "SingleModelCacheEntry",
|
3 |
+
"_model_id": "deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
|
4 |
+
"_task": "text-generation",
|
5 |
+
"architectures": [
|
6 |
+
"LlamaForCausalLM"
|
7 |
+
],
|
8 |
+
"attention_bias": false,
|
9 |
+
"attention_dropout": 0.0,
|
10 |
+
"head_dim": 128,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 8192,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 28672,
|
15 |
+
"max_position_embeddings": 131072,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"neuron": {
|
19 |
+
"_serialized_key": "NxDNeuronConfig",
|
20 |
+
"async_mode": false,
|
21 |
+
"attn_kernel_enabled": false,
|
22 |
+
"batch_size": 8,
|
23 |
+
"capacity_factor": null,
|
24 |
+
"cc_pipeline_tiling_factor": 2,
|
25 |
+
"checkpoint_id": "deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
|
26 |
+
"checkpoint_revision": "b1c0b44b4369b597ad119a196caf79a9c40e141e",
|
27 |
+
"continuous_batching": true,
|
28 |
+
"enable_bucketing": false,
|
29 |
+
"ep_degree": 1,
|
30 |
+
"flash_decoding_enabled": false,
|
31 |
+
"fused_qkv": true,
|
32 |
+
"glu_mlp": true,
|
33 |
+
"is_chunked_prefill": false,
|
34 |
+
"local_ranks_size": 24,
|
35 |
+
"logical_nc_config": 1,
|
36 |
+
"max_batch_size": 8,
|
37 |
+
"max_context_length": 4096,
|
38 |
+
"max_topk": 256,
|
39 |
+
"mlp_kernel_enabled": false,
|
40 |
+
"mlp_kernel_fuse_residual_add": false,
|
41 |
+
"n_active_tokens": 4096,
|
42 |
+
"neuronxcc_version": "2.17.194.0+d312836f",
|
43 |
+
"num_cores_per_group": 1,
|
44 |
+
"on_device_sampling": true,
|
45 |
+
"optimum_neuron_version": "0.2.0.dev7",
|
46 |
+
"output_logits": false,
|
47 |
+
"padding_side": "right",
|
48 |
+
"pp_degree": 1,
|
49 |
+
"qk_layernorm": false,
|
50 |
+
"qkv_kernel_enabled": false,
|
51 |
+
"rpl_reduce_dtype": "bfloat16",
|
52 |
+
"sequence_length": 4096,
|
53 |
+
"sequence_parallel_enabled": false,
|
54 |
+
"speculation_length": 0,
|
55 |
+
"start_rank_id": 0,
|
56 |
+
"target": null,
|
57 |
+
"torch_dtype": "bfloat16",
|
58 |
+
"tp_degree": 24,
|
59 |
+
"vocab_parallel": false
|
60 |
+
},
|
61 |
+
"num_attention_heads": 64,
|
62 |
+
"num_hidden_layers": 80,
|
63 |
+
"num_key_value_heads": 8,
|
64 |
+
"pretraining_tp": 1,
|
65 |
+
"rms_norm_eps": 1e-05,
|
66 |
+
"rope_scaling": {
|
67 |
+
"factor": 8.0,
|
68 |
+
"high_freq_factor": 4.0,
|
69 |
+
"low_freq_factor": 1.0,
|
70 |
+
"original_max_position_embeddings": 8192,
|
71 |
+
"rope_type": "llama3"
|
72 |
+
},
|
73 |
+
"rope_theta": 500000.0,
|
74 |
+
"tie_word_embeddings": false,
|
75 |
+
"use_cache": true,
|
76 |
+
"vocab_size": 128256
|
77 |
+
}
|
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/llamafactory/tiny-random-Llama-3/021f11fdefa8c3b516bd.json
ADDED
@@ -0,0 +1,77 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_entry_class": "SingleModelCacheEntry",
|
3 |
+
"_model_id": "llamafactory/tiny-random-Llama-3",
|
4 |
+
"_task": "text-generation",
|
5 |
+
"architectures": [
|
6 |
+
"LlamaForCausalLM"
|
7 |
+
],
|
8 |
+
"attention_bias": false,
|
9 |
+
"attention_dropout": 0.0,
|
10 |
+
"head_dim": 4,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 16,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 64,
|
15 |
+
"max_position_embeddings": 131072,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"neuron": {
|
19 |
+
"_serialized_key": "NxDNeuronConfig",
|
20 |
+
"async_mode": false,
|
21 |
+
"attn_kernel_enabled": false,
|
22 |
+
"batch_size": 2,
|
23 |
+
"capacity_factor": null,
|
24 |
+
"cc_pipeline_tiling_factor": 2,
|
25 |
+
"checkpoint_id": "llamafactory/tiny-random-Llama-3",
|
26 |
+
"checkpoint_revision": "bf2a2e3bf199ad2ee96f02a3c00246c608db22a8",
|
27 |
+
"continuous_batching": true,
|
28 |
+
"enable_bucketing": false,
|
29 |
+
"ep_degree": 1,
|
30 |
+
"flash_decoding_enabled": false,
|
31 |
+
"fused_qkv": true,
|
32 |
+
"glu_mlp": true,
|
33 |
+
"is_chunked_prefill": false,
|
34 |
+
"local_ranks_size": 2,
|
35 |
+
"logical_nc_config": 1,
|
36 |
+
"max_batch_size": 2,
|
37 |
+
"max_context_length": 128,
|
38 |
+
"max_topk": 256,
|
39 |
+
"mlp_kernel_enabled": false,
|
40 |
+
"mlp_kernel_fuse_residual_add": false,
|
41 |
+
"n_active_tokens": 128,
|
42 |
+
"neuronxcc_version": "2.17.194.0+d312836f",
|
43 |
+
"num_cores_per_group": 1,
|
44 |
+
"on_device_sampling": false,
|
45 |
+
"optimum_neuron_version": "0.2.0.dev7",
|
46 |
+
"output_logits": false,
|
47 |
+
"padding_side": "right",
|
48 |
+
"pp_degree": 1,
|
49 |
+
"qk_layernorm": false,
|
50 |
+
"qkv_kernel_enabled": false,
|
51 |
+
"rpl_reduce_dtype": "bfloat16",
|
52 |
+
"sequence_length": 128,
|
53 |
+
"sequence_parallel_enabled": false,
|
54 |
+
"speculation_length": 0,
|
55 |
+
"start_rank_id": 0,
|
56 |
+
"target": null,
|
57 |
+
"torch_dtype": "bfloat16",
|
58 |
+
"tp_degree": 2,
|
59 |
+
"vocab_parallel": false
|
60 |
+
},
|
61 |
+
"num_attention_heads": 4,
|
62 |
+
"num_hidden_layers": 2,
|
63 |
+
"num_key_value_heads": 4,
|
64 |
+
"pretraining_tp": 1,
|
65 |
+
"rms_norm_eps": 1e-05,
|
66 |
+
"rope_scaling": {
|
67 |
+
"factor": 8.0,
|
68 |
+
"high_freq_factor": 4.0,
|
69 |
+
"low_freq_factor": 1.0,
|
70 |
+
"original_max_position_embeddings": 8192,
|
71 |
+
"rope_type": "llama3"
|
72 |
+
},
|
73 |
+
"rope_theta": 500000.0,
|
74 |
+
"tie_word_embeddings": false,
|
75 |
+
"use_cache": true,
|
76 |
+
"vocab_size": 128256
|
77 |
+
}
|
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/llamafactory/tiny-random-Llama-3/1913cf3041a0fe975f3c.json
ADDED
@@ -0,0 +1,77 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_entry_class": "SingleModelCacheEntry",
|
3 |
+
"_model_id": "llamafactory/tiny-random-Llama-3",
|
4 |
+
"_task": "text-generation",
|
5 |
+
"architectures": [
|
6 |
+
"LlamaForCausalLM"
|
7 |
+
],
|
8 |
+
"attention_bias": false,
|
9 |
+
"attention_dropout": 0.0,
|
10 |
+
"head_dim": 4,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 16,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 64,
|
15 |
+
"max_position_embeddings": 131072,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"neuron": {
|
19 |
+
"_serialized_key": "NxDNeuronConfig",
|
20 |
+
"async_mode": false,
|
21 |
+
"attn_kernel_enabled": false,
|
22 |
+
"batch_size": 1,
|
23 |
+
"capacity_factor": null,
|
24 |
+
"cc_pipeline_tiling_factor": 2,
|
25 |
+
"checkpoint_id": "llamafactory/tiny-random-Llama-3",
|
26 |
+
"checkpoint_revision": "bf2a2e3bf199ad2ee96f02a3c00246c608db22a8",
|
27 |
+
"continuous_batching": false,
|
28 |
+
"enable_bucketing": false,
|
29 |
+
"ep_degree": 1,
|
30 |
+
"flash_decoding_enabled": false,
|
31 |
+
"fused_qkv": true,
|
32 |
+
"glu_mlp": true,
|
33 |
+
"is_chunked_prefill": false,
|
34 |
+
"local_ranks_size": 2,
|
35 |
+
"logical_nc_config": 1,
|
36 |
+
"max_batch_size": 1,
|
37 |
+
"max_context_length": 512,
|
38 |
+
"max_topk": 256,
|
39 |
+
"mlp_kernel_enabled": false,
|
40 |
+
"mlp_kernel_fuse_residual_add": false,
|
41 |
+
"n_active_tokens": 512,
|
42 |
+
"neuronxcc_version": "2.17.194.0+d312836f",
|
43 |
+
"num_cores_per_group": 1,
|
44 |
+
"on_device_sampling": true,
|
45 |
+
"optimum_neuron_version": "0.2.0.dev7",
|
46 |
+
"output_logits": false,
|
47 |
+
"padding_side": "right",
|
48 |
+
"pp_degree": 1,
|
49 |
+
"qk_layernorm": false,
|
50 |
+
"qkv_kernel_enabled": false,
|
51 |
+
"rpl_reduce_dtype": "float16",
|
52 |
+
"sequence_length": 512,
|
53 |
+
"sequence_parallel_enabled": false,
|
54 |
+
"speculation_length": 0,
|
55 |
+
"start_rank_id": 0,
|
56 |
+
"target": null,
|
57 |
+
"torch_dtype": "float16",
|
58 |
+
"tp_degree": 2,
|
59 |
+
"vocab_parallel": false
|
60 |
+
},
|
61 |
+
"num_attention_heads": 4,
|
62 |
+
"num_hidden_layers": 2,
|
63 |
+
"num_key_value_heads": 4,
|
64 |
+
"pretraining_tp": 1,
|
65 |
+
"rms_norm_eps": 1e-05,
|
66 |
+
"rope_scaling": {
|
67 |
+
"factor": 8.0,
|
68 |
+
"high_freq_factor": 4.0,
|
69 |
+
"low_freq_factor": 1.0,
|
70 |
+
"original_max_position_embeddings": 8192,
|
71 |
+
"rope_type": "llama3"
|
72 |
+
},
|
73 |
+
"rope_theta": 500000.0,
|
74 |
+
"tie_word_embeddings": false,
|
75 |
+
"use_cache": true,
|
76 |
+
"vocab_size": 128256
|
77 |
+
}
|
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/llamafactory/tiny-random-Llama-3/50411def20a2b703209e.json
ADDED
@@ -0,0 +1,77 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_entry_class": "SingleModelCacheEntry",
|
3 |
+
"_model_id": "llamafactory/tiny-random-Llama-3",
|
4 |
+
"_task": "text-generation",
|
5 |
+
"architectures": [
|
6 |
+
"LlamaForCausalLM"
|
7 |
+
],
|
8 |
+
"attention_bias": false,
|
9 |
+
"attention_dropout": 0.0,
|
10 |
+
"head_dim": 4,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 16,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 64,
|
15 |
+
"max_position_embeddings": 131072,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"neuron": {
|
19 |
+
"_serialized_key": "NxDNeuronConfig",
|
20 |
+
"async_mode": false,
|
21 |
+
"attn_kernel_enabled": false,
|
22 |
+
"batch_size": 1,
|
23 |
+
"capacity_factor": null,
|
24 |
+
"cc_pipeline_tiling_factor": 2,
|
25 |
+
"checkpoint_id": "llamafactory/tiny-random-Llama-3",
|
26 |
+
"checkpoint_revision": "bf2a2e3bf199ad2ee96f02a3c00246c608db22a8",
|
27 |
+
"continuous_batching": false,
|
28 |
+
"enable_bucketing": false,
|
29 |
+
"ep_degree": 1,
|
30 |
+
"flash_decoding_enabled": false,
|
31 |
+
"fused_qkv": true,
|
32 |
+
"glu_mlp": true,
|
33 |
+
"is_chunked_prefill": false,
|
34 |
+
"local_ranks_size": 1,
|
35 |
+
"logical_nc_config": 1,
|
36 |
+
"max_batch_size": 1,
|
37 |
+
"max_context_length": 512,
|
38 |
+
"max_topk": 256,
|
39 |
+
"mlp_kernel_enabled": false,
|
40 |
+
"mlp_kernel_fuse_residual_add": false,
|
41 |
+
"n_active_tokens": 512,
|
42 |
+
"neuronxcc_version": "2.17.194.0+d312836f",
|
43 |
+
"num_cores_per_group": 1,
|
44 |
+
"on_device_sampling": true,
|
45 |
+
"optimum_neuron_version": "0.2.0.dev7",
|
46 |
+
"output_logits": false,
|
47 |
+
"padding_side": "right",
|
48 |
+
"pp_degree": 1,
|
49 |
+
"qk_layernorm": false,
|
50 |
+
"qkv_kernel_enabled": false,
|
51 |
+
"rpl_reduce_dtype": "float16",
|
52 |
+
"sequence_length": 512,
|
53 |
+
"sequence_parallel_enabled": false,
|
54 |
+
"speculation_length": 0,
|
55 |
+
"start_rank_id": 0,
|
56 |
+
"target": null,
|
57 |
+
"torch_dtype": "float16",
|
58 |
+
"tp_degree": 1,
|
59 |
+
"vocab_parallel": false
|
60 |
+
},
|
61 |
+
"num_attention_heads": 4,
|
62 |
+
"num_hidden_layers": 2,
|
63 |
+
"num_key_value_heads": 4,
|
64 |
+
"pretraining_tp": 1,
|
65 |
+
"rms_norm_eps": 1e-05,
|
66 |
+
"rope_scaling": {
|
67 |
+
"factor": 8.0,
|
68 |
+
"high_freq_factor": 4.0,
|
69 |
+
"low_freq_factor": 1.0,
|
70 |
+
"original_max_position_embeddings": 8192,
|
71 |
+
"rope_type": "llama3"
|
72 |
+
},
|
73 |
+
"rope_theta": 500000.0,
|
74 |
+
"tie_word_embeddings": false,
|
75 |
+
"use_cache": true,
|
76 |
+
"vocab_size": 128256
|
77 |
+
}
|
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/llamafactory/tiny-random-Llama-3/7bca5f2b6f1034c1fa71.json
ADDED
@@ -0,0 +1,77 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_entry_class": "SingleModelCacheEntry",
|
3 |
+
"_model_id": "llamafactory/tiny-random-Llama-3",
|
4 |
+
"_task": "text-generation",
|
5 |
+
"architectures": [
|
6 |
+
"LlamaForCausalLM"
|
7 |
+
],
|
8 |
+
"attention_bias": false,
|
9 |
+
"attention_dropout": 0.0,
|
10 |
+
"head_dim": 4,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 16,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 64,
|
15 |
+
"max_position_embeddings": 131072,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"neuron": {
|
19 |
+
"_serialized_key": "NxDNeuronConfig",
|
20 |
+
"async_mode": false,
|
21 |
+
"attn_kernel_enabled": false,
|
22 |
+
"batch_size": 2,
|
23 |
+
"capacity_factor": null,
|
24 |
+
"cc_pipeline_tiling_factor": 2,
|
25 |
+
"checkpoint_id": "llamafactory/tiny-random-Llama-3",
|
26 |
+
"checkpoint_revision": "bf2a2e3bf199ad2ee96f02a3c00246c608db22a8",
|
27 |
+
"continuous_batching": true,
|
28 |
+
"enable_bucketing": false,
|
29 |
+
"ep_degree": 1,
|
30 |
+
"flash_decoding_enabled": false,
|
31 |
+
"fused_qkv": true,
|
32 |
+
"glu_mlp": true,
|
33 |
+
"is_chunked_prefill": false,
|
34 |
+
"local_ranks_size": 1,
|
35 |
+
"logical_nc_config": 1,
|
36 |
+
"max_batch_size": 2,
|
37 |
+
"max_context_length": 128,
|
38 |
+
"max_topk": 256,
|
39 |
+
"mlp_kernel_enabled": false,
|
40 |
+
"mlp_kernel_fuse_residual_add": false,
|
41 |
+
"n_active_tokens": 128,
|
42 |
+
"neuronxcc_version": "2.17.194.0+d312836f",
|
43 |
+
"num_cores_per_group": 1,
|
44 |
+
"on_device_sampling": true,
|
45 |
+
"optimum_neuron_version": "0.2.0.dev7",
|
46 |
+
"output_logits": false,
|
47 |
+
"padding_side": "right",
|
48 |
+
"pp_degree": 1,
|
49 |
+
"qk_layernorm": false,
|
50 |
+
"qkv_kernel_enabled": false,
|
51 |
+
"rpl_reduce_dtype": "bfloat16",
|
52 |
+
"sequence_length": 128,
|
53 |
+
"sequence_parallel_enabled": false,
|
54 |
+
"speculation_length": 0,
|
55 |
+
"start_rank_id": 0,
|
56 |
+
"target": null,
|
57 |
+
"torch_dtype": "bfloat16",
|
58 |
+
"tp_degree": 1,
|
59 |
+
"vocab_parallel": false
|
60 |
+
},
|
61 |
+
"num_attention_heads": 4,
|
62 |
+
"num_hidden_layers": 2,
|
63 |
+
"num_key_value_heads": 4,
|
64 |
+
"pretraining_tp": 1,
|
65 |
+
"rms_norm_eps": 1e-05,
|
66 |
+
"rope_scaling": {
|
67 |
+
"factor": 8.0,
|
68 |
+
"high_freq_factor": 4.0,
|
69 |
+
"low_freq_factor": 1.0,
|
70 |
+
"original_max_position_embeddings": 8192,
|
71 |
+
"rope_type": "llama3"
|
72 |
+
},
|
73 |
+
"rope_theta": 500000.0,
|
74 |
+
"tie_word_embeddings": false,
|
75 |
+
"use_cache": true,
|
76 |
+
"vocab_size": 128256
|
77 |
+
}
|
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/meta-llama/Llama-3.2-1B-Instruct/34beaf9835b09fe53395.json
ADDED
@@ -0,0 +1,77 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_entry_class": "SingleModelCacheEntry",
|
3 |
+
"_model_id": "meta-llama/Llama-3.2-1B-Instruct",
|
4 |
+
"_task": "text-generation",
|
5 |
+
"architectures": [
|
6 |
+
"LlamaForCausalLM"
|
7 |
+
],
|
8 |
+
"attention_bias": false,
|
9 |
+
"attention_dropout": 0.0,
|
10 |
+
"head_dim": 64,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 2048,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 8192,
|
15 |
+
"max_position_embeddings": 131072,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"neuron": {
|
19 |
+
"_serialized_key": "NxDNeuronConfig",
|
20 |
+
"async_mode": false,
|
21 |
+
"attn_kernel_enabled": false,
|
22 |
+
"batch_size": 4,
|
23 |
+
"capacity_factor": null,
|
24 |
+
"cc_pipeline_tiling_factor": 2,
|
25 |
+
"checkpoint_id": "meta-llama/Llama-3.2-1B-Instruct",
|
26 |
+
"checkpoint_revision": null,
|
27 |
+
"continuous_batching": true,
|
28 |
+
"enable_bucketing": false,
|
29 |
+
"ep_degree": 1,
|
30 |
+
"flash_decoding_enabled": false,
|
31 |
+
"fused_qkv": true,
|
32 |
+
"glu_mlp": true,
|
33 |
+
"is_chunked_prefill": false,
|
34 |
+
"local_ranks_size": 2,
|
35 |
+
"logical_nc_config": 1,
|
36 |
+
"max_batch_size": 4,
|
37 |
+
"max_context_length": 4096,
|
38 |
+
"max_topk": 256,
|
39 |
+
"mlp_kernel_enabled": false,
|
40 |
+
"mlp_kernel_fuse_residual_add": false,
|
41 |
+
"n_active_tokens": 4096,
|
42 |
+
"neuronxcc_version": "2.17.194.0+d312836f",
|
43 |
+
"num_cores_per_group": 1,
|
44 |
+
"on_device_sampling": false,
|
45 |
+
"optimum_neuron_version": "0.2.0.dev7",
|
46 |
+
"output_logits": false,
|
47 |
+
"padding_side": "right",
|
48 |
+
"pp_degree": 1,
|
49 |
+
"qk_layernorm": false,
|
50 |
+
"qkv_kernel_enabled": false,
|
51 |
+
"rpl_reduce_dtype": "bfloat16",
|
52 |
+
"sequence_length": 4096,
|
53 |
+
"sequence_parallel_enabled": false,
|
54 |
+
"speculation_length": 0,
|
55 |
+
"start_rank_id": 0,
|
56 |
+
"target": null,
|
57 |
+
"torch_dtype": "bfloat16",
|
58 |
+
"tp_degree": 2,
|
59 |
+
"vocab_parallel": false
|
60 |
+
},
|
61 |
+
"num_attention_heads": 32,
|
62 |
+
"num_hidden_layers": 16,
|
63 |
+
"num_key_value_heads": 8,
|
64 |
+
"pretraining_tp": 1,
|
65 |
+
"rms_norm_eps": 1e-05,
|
66 |
+
"rope_scaling": {
|
67 |
+
"factor": 32.0,
|
68 |
+
"high_freq_factor": 4.0,
|
69 |
+
"low_freq_factor": 1.0,
|
70 |
+
"original_max_position_embeddings": 8192,
|
71 |
+
"rope_type": "llama3"
|
72 |
+
},
|
73 |
+
"rope_theta": 500000.0,
|
74 |
+
"tie_word_embeddings": true,
|
75 |
+
"use_cache": true,
|
76 |
+
"vocab_size": 128256
|
77 |
+
}
|
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/meta-llama/Llama-3.2-1B-Instruct/d1ea2689244397f649b7.json
ADDED
@@ -0,0 +1,77 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_entry_class": "SingleModelCacheEntry",
|
3 |
+
"_model_id": "meta-llama/Llama-3.2-1B-Instruct",
|
4 |
+
"_task": "text-generation",
|
5 |
+
"architectures": [
|
6 |
+
"LlamaForCausalLM"
|
7 |
+
],
|
8 |
+
"attention_bias": false,
|
9 |
+
"attention_dropout": 0.0,
|
10 |
+
"head_dim": 64,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 2048,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 8192,
|
15 |
+
"max_position_embeddings": 131072,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"neuron": {
|
19 |
+
"_serialized_key": "NxDNeuronConfig",
|
20 |
+
"async_mode": false,
|
21 |
+
"attn_kernel_enabled": false,
|
22 |
+
"batch_size": 1,
|
23 |
+
"capacity_factor": null,
|
24 |
+
"cc_pipeline_tiling_factor": 2,
|
25 |
+
"checkpoint_id": "meta-llama/Llama-3.2-1B-Instruct",
|
26 |
+
"checkpoint_revision": "9213176726f574b556790deb65791e0c5aa438b6",
|
27 |
+
"continuous_batching": false,
|
28 |
+
"enable_bucketing": false,
|
29 |
+
"ep_degree": 1,
|
30 |
+
"flash_decoding_enabled": false,
|
31 |
+
"fused_qkv": true,
|
32 |
+
"glu_mlp": true,
|
33 |
+
"is_chunked_prefill": false,
|
34 |
+
"local_ranks_size": 2,
|
35 |
+
"logical_nc_config": 1,
|
36 |
+
"max_batch_size": 1,
|
37 |
+
"max_context_length": 4096,
|
38 |
+
"max_topk": 256,
|
39 |
+
"mlp_kernel_enabled": false,
|
40 |
+
"mlp_kernel_fuse_residual_add": false,
|
41 |
+
"n_active_tokens": 4096,
|
42 |
+
"neuronxcc_version": "2.17.194.0+d312836f",
|
43 |
+
"num_cores_per_group": 1,
|
44 |
+
"on_device_sampling": true,
|
45 |
+
"optimum_neuron_version": "0.2.0.dev7",
|
46 |
+
"output_logits": false,
|
47 |
+
"padding_side": "right",
|
48 |
+
"pp_degree": 1,
|
49 |
+
"qk_layernorm": false,
|
50 |
+
"qkv_kernel_enabled": false,
|
51 |
+
"rpl_reduce_dtype": "bfloat16",
|
52 |
+
"sequence_length": 4096,
|
53 |
+
"sequence_parallel_enabled": false,
|
54 |
+
"speculation_length": 0,
|
55 |
+
"start_rank_id": 0,
|
56 |
+
"target": null,
|
57 |
+
"torch_dtype": "bfloat16",
|
58 |
+
"tp_degree": 2,
|
59 |
+
"vocab_parallel": false
|
60 |
+
},
|
61 |
+
"num_attention_heads": 32,
|
62 |
+
"num_hidden_layers": 16,
|
63 |
+
"num_key_value_heads": 8,
|
64 |
+
"pretraining_tp": 1,
|
65 |
+
"rms_norm_eps": 1e-05,
|
66 |
+
"rope_scaling": {
|
67 |
+
"factor": 32.0,
|
68 |
+
"high_freq_factor": 4.0,
|
69 |
+
"low_freq_factor": 1.0,
|
70 |
+
"original_max_position_embeddings": 8192,
|
71 |
+
"rope_type": "llama3"
|
72 |
+
},
|
73 |
+
"rope_theta": 500000.0,
|
74 |
+
"tie_word_embeddings": true,
|
75 |
+
"use_cache": true,
|
76 |
+
"vocab_size": 128256
|
77 |
+
}
|
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/unsloth/Llama-3.1-8B-Instruct/321051129e499a9d100e.json
ADDED
@@ -0,0 +1,78 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_entry_class": "SingleModelCacheEntry",
|
3 |
+
"_model_id": "unsloth/Llama-3.1-8B-Instruct",
|
4 |
+
"_task": "text-generation",
|
5 |
+
"architectures": [
|
6 |
+
"LlamaForCausalLM"
|
7 |
+
],
|
8 |
+
"attention_bias": false,
|
9 |
+
"attention_dropout": 0.0,
|
10 |
+
"head_dim": 128,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 4096,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 14336,
|
15 |
+
"max_position_embeddings": 131072,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"neuron": {
|
19 |
+
"_serialized_key": "NxDNeuronConfig",
|
20 |
+
"async_mode": false,
|
21 |
+
"attn_kernel_enabled": false,
|
22 |
+
"batch_size": 32,
|
23 |
+
"capacity_factor": null,
|
24 |
+
"cc_pipeline_tiling_factor": 2,
|
25 |
+
"checkpoint_id": "unsloth/Llama-3.1-8B-Instruct",
|
26 |
+
"checkpoint_revision": "4699cc75b550f9c6f3173fb80f4703b62d946aa5",
|
27 |
+
"continuous_batching": true,
|
28 |
+
"enable_bucketing": false,
|
29 |
+
"ep_degree": 1,
|
30 |
+
"flash_decoding_enabled": false,
|
31 |
+
"fused_qkv": true,
|
32 |
+
"glu_mlp": true,
|
33 |
+
"is_chunked_prefill": false,
|
34 |
+
"local_ranks_size": 8,
|
35 |
+
"logical_nc_config": 1,
|
36 |
+
"max_batch_size": 32,
|
37 |
+
"max_context_length": 4096,
|
38 |
+
"max_topk": 256,
|
39 |
+
"mlp_kernel_enabled": false,
|
40 |
+
"mlp_kernel_fuse_residual_add": false,
|
41 |
+
"n_active_tokens": 4096,
|
42 |
+
"neuronxcc_version": "2.17.194.0+d312836f",
|
43 |
+
"num_cores_per_group": 1,
|
44 |
+
"on_device_sampling": true,
|
45 |
+
"optimum_neuron_version": "0.2.0.dev7",
|
46 |
+
"output_logits": false,
|
47 |
+
"padding_side": "right",
|
48 |
+
"pp_degree": 1,
|
49 |
+
"qk_layernorm": false,
|
50 |
+
"qkv_kernel_enabled": false,
|
51 |
+
"rpl_reduce_dtype": "bfloat16",
|
52 |
+
"sequence_length": 4096,
|
53 |
+
"sequence_parallel_enabled": false,
|
54 |
+
"speculation_length": 0,
|
55 |
+
"start_rank_id": 0,
|
56 |
+
"target": null,
|
57 |
+
"torch_dtype": "bfloat16",
|
58 |
+
"tp_degree": 8,
|
59 |
+
"vocab_parallel": false
|
60 |
+
},
|
61 |
+
"num_attention_heads": 32,
|
62 |
+
"num_hidden_layers": 32,
|
63 |
+
"num_key_value_heads": 8,
|
64 |
+
"pretraining_tp": 1,
|
65 |
+
"rms_norm_eps": 1e-05,
|
66 |
+
"rope_scaling": {
|
67 |
+
"factor": 8.0,
|
68 |
+
"high_freq_factor": 4.0,
|
69 |
+
"low_freq_factor": 1.0,
|
70 |
+
"original_max_position_embeddings": 8192,
|
71 |
+
"rope_type": "llama3"
|
72 |
+
},
|
73 |
+
"rope_theta": 500000.0,
|
74 |
+
"tie_word_embeddings": false,
|
75 |
+
"unsloth_fixed": true,
|
76 |
+
"use_cache": true,
|
77 |
+
"vocab_size": 128256
|
78 |
+
}
|
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/unsloth/Llama-3.1-8B-Instruct/59e56e036f276aac27ec.json
ADDED
@@ -0,0 +1,78 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_entry_class": "SingleModelCacheEntry",
|
3 |
+
"_model_id": "unsloth/Llama-3.1-8B-Instruct",
|
4 |
+
"_task": "text-generation",
|
5 |
+
"architectures": [
|
6 |
+
"LlamaForCausalLM"
|
7 |
+
],
|
8 |
+
"attention_bias": false,
|
9 |
+
"attention_dropout": 0.0,
|
10 |
+
"head_dim": 128,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 4096,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 14336,
|
15 |
+
"max_position_embeddings": 131072,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"neuron": {
|
19 |
+
"_serialized_key": "NxDNeuronConfig",
|
20 |
+
"async_mode": false,
|
21 |
+
"attn_kernel_enabled": false,
|
22 |
+
"batch_size": 8,
|
23 |
+
"capacity_factor": null,
|
24 |
+
"cc_pipeline_tiling_factor": 2,
|
25 |
+
"checkpoint_id": "unsloth/Llama-3.1-8B-Instruct",
|
26 |
+
"checkpoint_revision": "4699cc75b550f9c6f3173fb80f4703b62d946aa5",
|
27 |
+
"continuous_batching": true,
|
28 |
+
"enable_bucketing": false,
|
29 |
+
"ep_degree": 1,
|
30 |
+
"flash_decoding_enabled": false,
|
31 |
+
"fused_qkv": true,
|
32 |
+
"glu_mlp": true,
|
33 |
+
"is_chunked_prefill": false,
|
34 |
+
"local_ranks_size": 8,
|
35 |
+
"logical_nc_config": 1,
|
36 |
+
"max_batch_size": 8,
|
37 |
+
"max_context_length": 4096,
|
38 |
+
"max_topk": 256,
|
39 |
+
"mlp_kernel_enabled": false,
|
40 |
+
"mlp_kernel_fuse_residual_add": false,
|
41 |
+
"n_active_tokens": 4096,
|
42 |
+
"neuronxcc_version": "2.17.194.0+d312836f",
|
43 |
+
"num_cores_per_group": 1,
|
44 |
+
"on_device_sampling": true,
|
45 |
+
"optimum_neuron_version": "0.2.0.dev7",
|
46 |
+
"output_logits": false,
|
47 |
+
"padding_side": "right",
|
48 |
+
"pp_degree": 1,
|
49 |
+
"qk_layernorm": false,
|
50 |
+
"qkv_kernel_enabled": false,
|
51 |
+
"rpl_reduce_dtype": "bfloat16",
|
52 |
+
"sequence_length": 4096,
|
53 |
+
"sequence_parallel_enabled": false,
|
54 |
+
"speculation_length": 0,
|
55 |
+
"start_rank_id": 0,
|
56 |
+
"target": null,
|
57 |
+
"torch_dtype": "bfloat16",
|
58 |
+
"tp_degree": 8,
|
59 |
+
"vocab_parallel": false
|
60 |
+
},
|
61 |
+
"num_attention_heads": 32,
|
62 |
+
"num_hidden_layers": 32,
|
63 |
+
"num_key_value_heads": 8,
|
64 |
+
"pretraining_tp": 1,
|
65 |
+
"rms_norm_eps": 1e-05,
|
66 |
+
"rope_scaling": {
|
67 |
+
"factor": 8.0,
|
68 |
+
"high_freq_factor": 4.0,
|
69 |
+
"low_freq_factor": 1.0,
|
70 |
+
"original_max_position_embeddings": 8192,
|
71 |
+
"rope_type": "llama3"
|
72 |
+
},
|
73 |
+
"rope_theta": 500000.0,
|
74 |
+
"tie_word_embeddings": false,
|
75 |
+
"unsloth_fixed": true,
|
76 |
+
"use_cache": true,
|
77 |
+
"vocab_size": 128256
|
78 |
+
}
|
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/unsloth/Llama-3.1-8B-Instruct/aed3ac4481c88779a26c.json
ADDED
@@ -0,0 +1,78 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_entry_class": "SingleModelCacheEntry",
|
3 |
+
"_model_id": "unsloth/Llama-3.1-8B-Instruct",
|
4 |
+
"_task": "text-generation",
|
5 |
+
"architectures": [
|
6 |
+
"LlamaForCausalLM"
|
7 |
+
],
|
8 |
+
"attention_bias": false,
|
9 |
+
"attention_dropout": 0.0,
|
10 |
+
"head_dim": 128,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 4096,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 14336,
|
15 |
+
"max_position_embeddings": 131072,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"neuron": {
|
19 |
+
"_serialized_key": "NxDNeuronConfig",
|
20 |
+
"async_mode": false,
|
21 |
+
"attn_kernel_enabled": false,
|
22 |
+
"batch_size": 48,
|
23 |
+
"capacity_factor": null,
|
24 |
+
"cc_pipeline_tiling_factor": 2,
|
25 |
+
"checkpoint_id": "unsloth/Llama-3.1-8B-Instruct",
|
26 |
+
"checkpoint_revision": "4699cc75b550f9c6f3173fb80f4703b62d946aa5",
|
27 |
+
"continuous_batching": true,
|
28 |
+
"enable_bucketing": false,
|
29 |
+
"ep_degree": 1,
|
30 |
+
"flash_decoding_enabled": false,
|
31 |
+
"fused_qkv": true,
|
32 |
+
"glu_mlp": true,
|
33 |
+
"is_chunked_prefill": false,
|
34 |
+
"local_ranks_size": 8,
|
35 |
+
"logical_nc_config": 1,
|
36 |
+
"max_batch_size": 48,
|
37 |
+
"max_context_length": 4096,
|
38 |
+
"max_topk": 256,
|
39 |
+
"mlp_kernel_enabled": false,
|
40 |
+
"mlp_kernel_fuse_residual_add": false,
|
41 |
+
"n_active_tokens": 4096,
|
42 |
+
"neuronxcc_version": "2.17.194.0+d312836f",
|
43 |
+
"num_cores_per_group": 1,
|
44 |
+
"on_device_sampling": true,
|
45 |
+
"optimum_neuron_version": "0.2.0.dev7",
|
46 |
+
"output_logits": false,
|
47 |
+
"padding_side": "right",
|
48 |
+
"pp_degree": 1,
|
49 |
+
"qk_layernorm": false,
|
50 |
+
"qkv_kernel_enabled": false,
|
51 |
+
"rpl_reduce_dtype": "bfloat16",
|
52 |
+
"sequence_length": 4096,
|
53 |
+
"sequence_parallel_enabled": false,
|
54 |
+
"speculation_length": 0,
|
55 |
+
"start_rank_id": 0,
|
56 |
+
"target": null,
|
57 |
+
"torch_dtype": "bfloat16",
|
58 |
+
"tp_degree": 8,
|
59 |
+
"vocab_parallel": false
|
60 |
+
},
|
61 |
+
"num_attention_heads": 32,
|
62 |
+
"num_hidden_layers": 32,
|
63 |
+
"num_key_value_heads": 8,
|
64 |
+
"pretraining_tp": 1,
|
65 |
+
"rms_norm_eps": 1e-05,
|
66 |
+
"rope_scaling": {
|
67 |
+
"factor": 8.0,
|
68 |
+
"high_freq_factor": 4.0,
|
69 |
+
"low_freq_factor": 1.0,
|
70 |
+
"original_max_position_embeddings": 8192,
|
71 |
+
"rope_type": "llama3"
|
72 |
+
},
|
73 |
+
"rope_theta": 500000.0,
|
74 |
+
"tie_word_embeddings": false,
|
75 |
+
"unsloth_fixed": true,
|
76 |
+
"use_cache": true,
|
77 |
+
"vocab_size": 128256
|
78 |
+
}
|
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/unsloth/Llama-3.1-8B-Instruct/bcd80b4d12e05bc045ce.json
ADDED
@@ -0,0 +1,78 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_entry_class": "SingleModelCacheEntry",
|
3 |
+
"_model_id": "unsloth/Llama-3.1-8B-Instruct",
|
4 |
+
"_task": "text-generation",
|
5 |
+
"architectures": [
|
6 |
+
"LlamaForCausalLM"
|
7 |
+
],
|
8 |
+
"attention_bias": false,
|
9 |
+
"attention_dropout": 0.0,
|
10 |
+
"head_dim": 128,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 4096,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 14336,
|
15 |
+
"max_position_embeddings": 131072,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"neuron": {
|
19 |
+
"_serialized_key": "NxDNeuronConfig",
|
20 |
+
"async_mode": false,
|
21 |
+
"attn_kernel_enabled": false,
|
22 |
+
"batch_size": 64,
|
23 |
+
"capacity_factor": null,
|
24 |
+
"cc_pipeline_tiling_factor": 2,
|
25 |
+
"checkpoint_id": "unsloth/Llama-3.1-8B-Instruct",
|
26 |
+
"checkpoint_revision": "4699cc75b550f9c6f3173fb80f4703b62d946aa5",
|
27 |
+
"continuous_batching": true,
|
28 |
+
"enable_bucketing": false,
|
29 |
+
"ep_degree": 1,
|
30 |
+
"flash_decoding_enabled": false,
|
31 |
+
"fused_qkv": true,
|
32 |
+
"glu_mlp": true,
|
33 |
+
"is_chunked_prefill": false,
|
34 |
+
"local_ranks_size": 8,
|
35 |
+
"logical_nc_config": 1,
|
36 |
+
"max_batch_size": 64,
|
37 |
+
"max_context_length": 4096,
|
38 |
+
"max_topk": 256,
|
39 |
+
"mlp_kernel_enabled": false,
|
40 |
+
"mlp_kernel_fuse_residual_add": false,
|
41 |
+
"n_active_tokens": 4096,
|
42 |
+
"neuronxcc_version": "2.17.194.0+d312836f",
|
43 |
+
"num_cores_per_group": 1,
|
44 |
+
"on_device_sampling": true,
|
45 |
+
"optimum_neuron_version": "0.2.0.dev7",
|
46 |
+
"output_logits": false,
|
47 |
+
"padding_side": "right",
|
48 |
+
"pp_degree": 1,
|
49 |
+
"qk_layernorm": false,
|
50 |
+
"qkv_kernel_enabled": false,
|
51 |
+
"rpl_reduce_dtype": "bfloat16",
|
52 |
+
"sequence_length": 4096,
|
53 |
+
"sequence_parallel_enabled": false,
|
54 |
+
"speculation_length": 0,
|
55 |
+
"start_rank_id": 0,
|
56 |
+
"target": null,
|
57 |
+
"torch_dtype": "bfloat16",
|
58 |
+
"tp_degree": 8,
|
59 |
+
"vocab_parallel": false
|
60 |
+
},
|
61 |
+
"num_attention_heads": 32,
|
62 |
+
"num_hidden_layers": 32,
|
63 |
+
"num_key_value_heads": 8,
|
64 |
+
"pretraining_tp": 1,
|
65 |
+
"rms_norm_eps": 1e-05,
|
66 |
+
"rope_scaling": {
|
67 |
+
"factor": 8.0,
|
68 |
+
"high_freq_factor": 4.0,
|
69 |
+
"low_freq_factor": 1.0,
|
70 |
+
"original_max_position_embeddings": 8192,
|
71 |
+
"rope_type": "llama3"
|
72 |
+
},
|
73 |
+
"rope_theta": 500000.0,
|
74 |
+
"tie_word_embeddings": false,
|
75 |
+
"unsloth_fixed": true,
|
76 |
+
"use_cache": true,
|
77 |
+
"vocab_size": 128256
|
78 |
+
}
|
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/unsloth/Llama-3.1-8B-Instruct/c28ab0c7d33e28708b3c.json
ADDED
@@ -0,0 +1,78 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_entry_class": "SingleModelCacheEntry",
|
3 |
+
"_model_id": "unsloth/Llama-3.1-8B-Instruct",
|
4 |
+
"_task": "text-generation",
|
5 |
+
"architectures": [
|
6 |
+
"LlamaForCausalLM"
|
7 |
+
],
|
8 |
+
"attention_bias": false,
|
9 |
+
"attention_dropout": 0.0,
|
10 |
+
"head_dim": 128,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 4096,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 14336,
|
15 |
+
"max_position_embeddings": 131072,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"neuron": {
|
19 |
+
"_serialized_key": "NxDNeuronConfig",
|
20 |
+
"async_mode": false,
|
21 |
+
"attn_kernel_enabled": false,
|
22 |
+
"batch_size": 1,
|
23 |
+
"capacity_factor": null,
|
24 |
+
"cc_pipeline_tiling_factor": 2,
|
25 |
+
"checkpoint_id": "unsloth/Llama-3.1-8B-Instruct",
|
26 |
+
"checkpoint_revision": "4699cc75b550f9c6f3173fb80f4703b62d946aa5",
|
27 |
+
"continuous_batching": false,
|
28 |
+
"enable_bucketing": false,
|
29 |
+
"ep_degree": 1,
|
30 |
+
"flash_decoding_enabled": false,
|
31 |
+
"fused_qkv": true,
|
32 |
+
"glu_mlp": true,
|
33 |
+
"is_chunked_prefill": false,
|
34 |
+
"local_ranks_size": 8,
|
35 |
+
"logical_nc_config": 1,
|
36 |
+
"max_batch_size": 1,
|
37 |
+
"max_context_length": 4096,
|
38 |
+
"max_topk": 256,
|
39 |
+
"mlp_kernel_enabled": false,
|
40 |
+
"mlp_kernel_fuse_residual_add": false,
|
41 |
+
"n_active_tokens": 4096,
|
42 |
+
"neuronxcc_version": "2.17.194.0+d312836f",
|
43 |
+
"num_cores_per_group": 1,
|
44 |
+
"on_device_sampling": true,
|
45 |
+
"optimum_neuron_version": "0.2.0.dev7",
|
46 |
+
"output_logits": false,
|
47 |
+
"padding_side": "right",
|
48 |
+
"pp_degree": 1,
|
49 |
+
"qk_layernorm": false,
|
50 |
+
"qkv_kernel_enabled": false,
|
51 |
+
"rpl_reduce_dtype": "bfloat16",
|
52 |
+
"sequence_length": 4096,
|
53 |
+
"sequence_parallel_enabled": false,
|
54 |
+
"speculation_length": 0,
|
55 |
+
"start_rank_id": 0,
|
56 |
+
"target": null,
|
57 |
+
"torch_dtype": "bfloat16",
|
58 |
+
"tp_degree": 8,
|
59 |
+
"vocab_parallel": false
|
60 |
+
},
|
61 |
+
"num_attention_heads": 32,
|
62 |
+
"num_hidden_layers": 32,
|
63 |
+
"num_key_value_heads": 8,
|
64 |
+
"pretraining_tp": 1,
|
65 |
+
"rms_norm_eps": 1e-05,
|
66 |
+
"rope_scaling": {
|
67 |
+
"factor": 8.0,
|
68 |
+
"high_freq_factor": 4.0,
|
69 |
+
"low_freq_factor": 1.0,
|
70 |
+
"original_max_position_embeddings": 8192,
|
71 |
+
"rope_type": "llama3"
|
72 |
+
},
|
73 |
+
"rope_theta": 500000.0,
|
74 |
+
"tie_word_embeddings": false,
|
75 |
+
"unsloth_fixed": true,
|
76 |
+
"use_cache": true,
|
77 |
+
"vocab_size": 128256
|
78 |
+
}
|
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/unsloth/Llama-3.1-8B-Instruct/f0e18f873ce42aaa0b9d.json
ADDED
@@ -0,0 +1,78 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_entry_class": "SingleModelCacheEntry",
|
3 |
+
"_model_id": "unsloth/Llama-3.1-8B-Instruct",
|
4 |
+
"_task": "text-generation",
|
5 |
+
"architectures": [
|
6 |
+
"LlamaForCausalLM"
|
7 |
+
],
|
8 |
+
"attention_bias": false,
|
9 |
+
"attention_dropout": 0.0,
|
10 |
+
"head_dim": 128,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 4096,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 14336,
|
15 |
+
"max_position_embeddings": 131072,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"neuron": {
|
19 |
+
"_serialized_key": "NxDNeuronConfig",
|
20 |
+
"async_mode": false,
|
21 |
+
"attn_kernel_enabled": false,
|
22 |
+
"batch_size": 4,
|
23 |
+
"capacity_factor": null,
|
24 |
+
"cc_pipeline_tiling_factor": 2,
|
25 |
+
"checkpoint_id": "unsloth/Llama-3.1-8B-Instruct",
|
26 |
+
"checkpoint_revision": "4699cc75b550f9c6f3173fb80f4703b62d946aa5",
|
27 |
+
"continuous_batching": true,
|
28 |
+
"enable_bucketing": false,
|
29 |
+
"ep_degree": 1,
|
30 |
+
"flash_decoding_enabled": false,
|
31 |
+
"fused_qkv": true,
|
32 |
+
"glu_mlp": true,
|
33 |
+
"is_chunked_prefill": false,
|
34 |
+
"local_ranks_size": 8,
|
35 |
+
"logical_nc_config": 1,
|
36 |
+
"max_batch_size": 4,
|
37 |
+
"max_context_length": 4096,
|
38 |
+
"max_topk": 256,
|
39 |
+
"mlp_kernel_enabled": false,
|
40 |
+
"mlp_kernel_fuse_residual_add": false,
|
41 |
+
"n_active_tokens": 4096,
|
42 |
+
"neuronxcc_version": "2.17.194.0+d312836f",
|
43 |
+
"num_cores_per_group": 1,
|
44 |
+
"on_device_sampling": true,
|
45 |
+
"optimum_neuron_version": "0.2.0.dev7",
|
46 |
+
"output_logits": false,
|
47 |
+
"padding_side": "right",
|
48 |
+
"pp_degree": 1,
|
49 |
+
"qk_layernorm": false,
|
50 |
+
"qkv_kernel_enabled": false,
|
51 |
+
"rpl_reduce_dtype": "bfloat16",
|
52 |
+
"sequence_length": 4096,
|
53 |
+
"sequence_parallel_enabled": false,
|
54 |
+
"speculation_length": 0,
|
55 |
+
"start_rank_id": 0,
|
56 |
+
"target": null,
|
57 |
+
"torch_dtype": "bfloat16",
|
58 |
+
"tp_degree": 8,
|
59 |
+
"vocab_parallel": false
|
60 |
+
},
|
61 |
+
"num_attention_heads": 32,
|
62 |
+
"num_hidden_layers": 32,
|
63 |
+
"num_key_value_heads": 8,
|
64 |
+
"pretraining_tp": 1,
|
65 |
+
"rms_norm_eps": 1e-05,
|
66 |
+
"rope_scaling": {
|
67 |
+
"factor": 8.0,
|
68 |
+
"high_freq_factor": 4.0,
|
69 |
+
"low_freq_factor": 1.0,
|
70 |
+
"original_max_position_embeddings": 8192,
|
71 |
+
"rope_type": "llama3"
|
72 |
+
},
|
73 |
+
"rope_theta": 500000.0,
|
74 |
+
"tie_word_embeddings": false,
|
75 |
+
"unsloth_fixed": true,
|
76 |
+
"use_cache": true,
|
77 |
+
"vocab_size": 128256
|
78 |
+
}
|
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev7/llama/unsloth/Llama-3.1-8B-Instruct/fb1938af2d9e7e083207.json
ADDED
@@ -0,0 +1,78 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_entry_class": "SingleModelCacheEntry",
|
3 |
+
"_model_id": "unsloth/Llama-3.1-8B-Instruct",
|
4 |
+
"_task": "text-generation",
|
5 |
+
"architectures": [
|
6 |
+
"LlamaForCausalLM"
|
7 |
+
],
|
8 |
+
"attention_bias": false,
|
9 |
+
"attention_dropout": 0.0,
|
10 |
+
"head_dim": 128,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 4096,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 14336,
|
15 |
+
"max_position_embeddings": 131072,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"neuron": {
|
19 |
+
"_serialized_key": "NxDNeuronConfig",
|
20 |
+
"async_mode": false,
|
21 |
+
"attn_kernel_enabled": false,
|
22 |
+
"batch_size": 16,
|
23 |
+
"capacity_factor": null,
|
24 |
+
"cc_pipeline_tiling_factor": 2,
|
25 |
+
"checkpoint_id": "unsloth/Llama-3.1-8B-Instruct",
|
26 |
+
"checkpoint_revision": "4699cc75b550f9c6f3173fb80f4703b62d946aa5",
|
27 |
+
"continuous_batching": true,
|
28 |
+
"enable_bucketing": false,
|
29 |
+
"ep_degree": 1,
|
30 |
+
"flash_decoding_enabled": false,
|
31 |
+
"fused_qkv": true,
|
32 |
+
"glu_mlp": true,
|
33 |
+
"is_chunked_prefill": false,
|
34 |
+
"local_ranks_size": 8,
|
35 |
+
"logical_nc_config": 1,
|
36 |
+
"max_batch_size": 16,
|
37 |
+
"max_context_length": 4096,
|
38 |
+
"max_topk": 256,
|
39 |
+
"mlp_kernel_enabled": false,
|
40 |
+
"mlp_kernel_fuse_residual_add": false,
|
41 |
+
"n_active_tokens": 4096,
|
42 |
+
"neuronxcc_version": "2.17.194.0+d312836f",
|
43 |
+
"num_cores_per_group": 1,
|
44 |
+
"on_device_sampling": true,
|
45 |
+
"optimum_neuron_version": "0.2.0.dev7",
|
46 |
+
"output_logits": false,
|
47 |
+
"padding_side": "right",
|
48 |
+
"pp_degree": 1,
|
49 |
+
"qk_layernorm": false,
|
50 |
+
"qkv_kernel_enabled": false,
|
51 |
+
"rpl_reduce_dtype": "bfloat16",
|
52 |
+
"sequence_length": 4096,
|
53 |
+
"sequence_parallel_enabled": false,
|
54 |
+
"speculation_length": 0,
|
55 |
+
"start_rank_id": 0,
|
56 |
+
"target": null,
|
57 |
+
"torch_dtype": "bfloat16",
|
58 |
+
"tp_degree": 8,
|
59 |
+
"vocab_parallel": false
|
60 |
+
},
|
61 |
+
"num_attention_heads": 32,
|
62 |
+
"num_hidden_layers": 32,
|
63 |
+
"num_key_value_heads": 8,
|
64 |
+
"pretraining_tp": 1,
|
65 |
+
"rms_norm_eps": 1e-05,
|
66 |
+
"rope_scaling": {
|
67 |
+
"factor": 8.0,
|
68 |
+
"high_freq_factor": 4.0,
|
69 |
+
"low_freq_factor": 1.0,
|
70 |
+
"original_max_position_embeddings": 8192,
|
71 |
+
"rope_type": "llama3"
|
72 |
+
},
|
73 |
+
"rope_theta": 500000.0,
|
74 |
+
"tie_word_embeddings": false,
|
75 |
+
"unsloth_fixed": true,
|
76 |
+
"use_cache": true,
|
77 |
+
"vocab_size": 128256
|
78 |
+
}
|
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev8/llama/meta-llama/Llama-3.2-1B-Instruct/3da196f521260e769ca6.json
ADDED
@@ -0,0 +1,77 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_entry_class": "SingleModelCacheEntry",
|
3 |
+
"_model_id": "meta-llama/Llama-3.2-1B-Instruct",
|
4 |
+
"_task": "text-generation",
|
5 |
+
"architectures": [
|
6 |
+
"LlamaForCausalLM"
|
7 |
+
],
|
8 |
+
"attention_bias": false,
|
9 |
+
"attention_dropout": 0.0,
|
10 |
+
"head_dim": 64,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 2048,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 8192,
|
15 |
+
"max_position_embeddings": 131072,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"neuron": {
|
19 |
+
"_serialized_key": "NxDNeuronConfig",
|
20 |
+
"async_mode": false,
|
21 |
+
"attn_kernel_enabled": false,
|
22 |
+
"batch_size": 1,
|
23 |
+
"capacity_factor": null,
|
24 |
+
"cc_pipeline_tiling_factor": 2,
|
25 |
+
"checkpoint_id": "meta-llama/Llama-3.2-1B-Instruct",
|
26 |
+
"checkpoint_revision": "9213176726f574b556790deb65791e0c5aa438b6",
|
27 |
+
"continuous_batching": false,
|
28 |
+
"enable_bucketing": false,
|
29 |
+
"ep_degree": 1,
|
30 |
+
"flash_decoding_enabled": false,
|
31 |
+
"fused_qkv": true,
|
32 |
+
"glu_mlp": true,
|
33 |
+
"is_chunked_prefill": false,
|
34 |
+
"local_ranks_size": 2,
|
35 |
+
"logical_nc_config": 1,
|
36 |
+
"max_batch_size": 1,
|
37 |
+
"max_context_length": 4096,
|
38 |
+
"max_topk": 256,
|
39 |
+
"mlp_kernel_enabled": false,
|
40 |
+
"mlp_kernel_fuse_residual_add": false,
|
41 |
+
"n_active_tokens": 4096,
|
42 |
+
"neuronxcc_version": "2.17.194.0+d312836f",
|
43 |
+
"num_cores_per_group": 1,
|
44 |
+
"on_device_sampling": true,
|
45 |
+
"optimum_neuron_version": "0.2.0.dev8",
|
46 |
+
"output_logits": false,
|
47 |
+
"padding_side": "right",
|
48 |
+
"pp_degree": 1,
|
49 |
+
"qk_layernorm": false,
|
50 |
+
"qkv_kernel_enabled": false,
|
51 |
+
"rpl_reduce_dtype": "bfloat16",
|
52 |
+
"sequence_length": 4096,
|
53 |
+
"sequence_parallel_enabled": false,
|
54 |
+
"speculation_length": 0,
|
55 |
+
"start_rank_id": 0,
|
56 |
+
"target": null,
|
57 |
+
"torch_dtype": "bfloat16",
|
58 |
+
"tp_degree": 2,
|
59 |
+
"vocab_parallel": false
|
60 |
+
},
|
61 |
+
"num_attention_heads": 32,
|
62 |
+
"num_hidden_layers": 16,
|
63 |
+
"num_key_value_heads": 8,
|
64 |
+
"pretraining_tp": 1,
|
65 |
+
"rms_norm_eps": 1e-05,
|
66 |
+
"rope_scaling": {
|
67 |
+
"factor": 32.0,
|
68 |
+
"high_freq_factor": 4.0,
|
69 |
+
"low_freq_factor": 1.0,
|
70 |
+
"original_max_position_embeddings": 8192,
|
71 |
+
"rope_type": "llama3"
|
72 |
+
},
|
73 |
+
"rope_theta": 500000.0,
|
74 |
+
"tie_word_embeddings": true,
|
75 |
+
"use_cache": true,
|
76 |
+
"vocab_size": 128256
|
77 |
+
}
|
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.0.dev8/llama/unsloth/Llama-3.2-1B-Instruct/f4d3009ec7e739e5ca73.json
ADDED
@@ -0,0 +1,78 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_entry_class": "SingleModelCacheEntry",
|
3 |
+
"_model_id": "unsloth/Llama-3.2-1B-Instruct",
|
4 |
+
"_task": "text-generation",
|
5 |
+
"architectures": [
|
6 |
+
"LlamaForCausalLM"
|
7 |
+
],
|
8 |
+
"attention_bias": false,
|
9 |
+
"attention_dropout": 0.0,
|
10 |
+
"head_dim": 64,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 2048,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 8192,
|
15 |
+
"max_position_embeddings": 131072,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"neuron": {
|
19 |
+
"_serialized_key": "NxDNeuronConfig",
|
20 |
+
"async_mode": false,
|
21 |
+
"attn_kernel_enabled": false,
|
22 |
+
"batch_size": 4,
|
23 |
+
"capacity_factor": null,
|
24 |
+
"cc_pipeline_tiling_factor": 2,
|
25 |
+
"checkpoint_id": "unsloth/Llama-3.2-1B-Instruct",
|
26 |
+
"checkpoint_revision": "5a8abab4a5d6f164389b1079fb721cfab8d7126c",
|
27 |
+
"continuous_batching": true,
|
28 |
+
"enable_bucketing": false,
|
29 |
+
"ep_degree": 1,
|
30 |
+
"flash_decoding_enabled": false,
|
31 |
+
"fused_qkv": true,
|
32 |
+
"glu_mlp": true,
|
33 |
+
"is_chunked_prefill": false,
|
34 |
+
"local_ranks_size": 2,
|
35 |
+
"logical_nc_config": 1,
|
36 |
+
"max_batch_size": 4,
|
37 |
+
"max_context_length": 4096,
|
38 |
+
"max_topk": 256,
|
39 |
+
"mlp_kernel_enabled": false,
|
40 |
+
"mlp_kernel_fuse_residual_add": false,
|
41 |
+
"n_active_tokens": 4096,
|
42 |
+
"neuronxcc_version": "2.17.194.0+d312836f",
|
43 |
+
"num_cores_per_group": 1,
|
44 |
+
"on_device_sampling": false,
|
45 |
+
"optimum_neuron_version": "0.2.0.dev8",
|
46 |
+
"output_logits": false,
|
47 |
+
"padding_side": "right",
|
48 |
+
"pp_degree": 1,
|
49 |
+
"qk_layernorm": false,
|
50 |
+
"qkv_kernel_enabled": false,
|
51 |
+
"rpl_reduce_dtype": "float16",
|
52 |
+
"sequence_length": 4096,
|
53 |
+
"sequence_parallel_enabled": false,
|
54 |
+
"speculation_length": 0,
|
55 |
+
"start_rank_id": 0,
|
56 |
+
"target": null,
|
57 |
+
"torch_dtype": "float16",
|
58 |
+
"tp_degree": 2,
|
59 |
+
"vocab_parallel": false
|
60 |
+
},
|
61 |
+
"num_attention_heads": 32,
|
62 |
+
"num_hidden_layers": 16,
|
63 |
+
"num_key_value_heads": 8,
|
64 |
+
"pretraining_tp": 1,
|
65 |
+
"rms_norm_eps": 1e-05,
|
66 |
+
"rope_scaling": {
|
67 |
+
"factor": 32.0,
|
68 |
+
"high_freq_factor": 4.0,
|
69 |
+
"low_freq_factor": 1.0,
|
70 |
+
"original_max_position_embeddings": 8192,
|
71 |
+
"rope_type": "llama3"
|
72 |
+
},
|
73 |
+
"rope_theta": 500000.0,
|
74 |
+
"tie_word_embeddings": true,
|
75 |
+
"unsloth_fixed": true,
|
76 |
+
"use_cache": true,
|
77 |
+
"vocab_size": 128256
|
78 |
+
}
|
neuronxcc-2.17.194.0+d312836f/MODULE_056a0c30057367f9353e+165e9558/compile_flags.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
"--auto-cast=none --model-type=transformer --tensorizer-options='--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-dge-dma --vectorize-strided-dma ' -O2 --internal-num-neuroncores-per-sengine=1 --logfile=/tmp/nxd_model/token_generation_model/_tp0_bk0/log-neuron-cc.txt --enable-internal-neff-wrapper"
|
neuronxcc-2.17.194.0+d312836f/MODULE_056a0c30057367f9353e+165e9558/model.done
ADDED
File without changes
|
neuronxcc-2.17.194.0+d312836f/MODULE_056a0c30057367f9353e+165e9558/model.hlo_module.pb
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:16b6a6b4a1369a831979e2c28c208afbf421b340891cba487bb8c6d49511d31a
|
3 |
+
size 1915034
|
neuronxcc-2.17.194.0+d312836f/MODULE_056a0c30057367f9353e+165e9558/model.neff
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2aaf3fd8a2d76bb58beb1ffcd2c98841e8eb0164b432ad31f3066ac6494cd1dc
|
3 |
+
size 11787264
|
neuronxcc-2.17.194.0+d312836f/MODULE_056a0c30057367f9353e+165e9558/wrapped_neff.hlo
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5a9d6cf5b5ca0842341498a231fd41087cc6df62fe6d04e8b924611073b418f5
|
3 |
+
size 12130962
|
neuronxcc-2.17.194.0+d312836f/MODULE_07e4a454b533bd5b9f6f+bfe5714b/compile_flags.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
"--auto-cast=none --model-type=transformer --tensorizer-options='--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-dge-dma --vectorize-strided-dma ' -O2 --internal-num-neuroncores-per-sengine=1 --logfile=/tmp/nxd_model/context_encoding_model/_tp0_bk0/log-neuron-cc.txt"
|
neuronxcc-2.17.194.0+d312836f/MODULE_07e4a454b533bd5b9f6f+bfe5714b/model.done
ADDED
File without changes
|
neuronxcc-2.17.194.0+d312836f/MODULE_07e4a454b533bd5b9f6f+bfe5714b/model.hlo_module.pb
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7b4e39ff7264047e564f306270e3be1e754a5c0edc58e50acbff1bb553452b53
|
3 |
+
size 920875
|
neuronxcc-2.17.194.0+d312836f/MODULE_07e4a454b533bd5b9f6f+bfe5714b/model.neff
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:54c6db91e68c2de260b3d051da59a240cf5afc202104e4fcf45f4ba09cc6bd59
|
3 |
+
size 32646144
|
neuronxcc-2.17.194.0+d312836f/MODULE_08eb5012506859e6b129+165e9558/compile_flags.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
"--auto-cast=none --model-type=transformer --tensorizer-options='--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-dge-dma --vectorize-strided-dma ' -O2 --internal-num-neuroncores-per-sengine=1 --logfile=/tmp/nxd_model/token_generation_model/_tp0_bk0/log-neuron-cc.txt --enable-internal-neff-wrapper"
|
neuronxcc-2.17.194.0+d312836f/MODULE_08eb5012506859e6b129+165e9558/model.done
ADDED
File without changes
|
neuronxcc-2.17.194.0+d312836f/MODULE_08eb5012506859e6b129+165e9558/model.hlo_module.pb
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e23333e0184c701665b77443dff9e0e7826cde3df7b3a8610dfd7a6cb3bf1916
|
3 |
+
size 79913
|
neuronxcc-2.17.194.0+d312836f/MODULE_08eb5012506859e6b129+165e9558/model.neff
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:932fc04b510efe964eaa7ba1983cedcd49d5bc2fb595b9b914ab4a3f69185d6c
|
3 |
+
size 236544
|
neuronxcc-2.17.194.0+d312836f/MODULE_08eb5012506859e6b129+165e9558/wrapped_neff.hlo
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9eb326a41ddb5d6d9473b240905d4d1e8cb2900bada8756712c4a03aea93ee03
|
3 |
+
size 244351
|
neuronxcc-2.17.194.0+d312836f/MODULE_184d4ed11977011ce5e7+431f5505/compile_flags.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
"--model-type=transformer -O1 --lnc=1 --internal-hlo2tensorizer-options=--experimental-unsafe-fp8e4m3fn-as-fp8e4m3 --logfile=/tmp/nxd_model/layout_opt/log-neuron-cc.txt"
|
neuronxcc-2.17.194.0+d312836f/MODULE_184d4ed11977011ce5e7+431f5505/model.done
ADDED
File without changes
|
neuronxcc-2.17.194.0+d312836f/MODULE_184d4ed11977011ce5e7+431f5505/model.hlo_module.pb
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d5928e1e3d71165915a25d2d9778aedfc58cd57ef9ba8621279558be58e16fb7
|
3 |
+
size 136713
|
neuronxcc-2.17.194.0+d312836f/MODULE_184d4ed11977011ce5e7+431f5505/model.neff
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5b15ec188343dcedabc48667e269d3cafe56b8fcb199b8eed4cabf83ab5851d6
|
3 |
+
size 2223104
|
neuronxcc-2.17.194.0+d312836f/MODULE_24ff9ac2787ce9a1d276+613edded/model.neff
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 134144
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8ed82a5211308e005fecaac2b616b3b285f234a482ccd5ac1a1be4e47c53f2fc
|
3 |
size 134144
|
neuronxcc-2.17.194.0+d312836f/MODULE_320f2622d4d0c9fdd0f1+613edded/model.neff
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 144384
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0b6bf86580e24a1eb275a27ab3a061cc5eb70cbaee41a624f938f0c23833e3fd
|
3 |
size 144384
|
neuronxcc-2.17.194.0+d312836f/MODULE_374002f5e545876dbed9+bfe5714b/compile_flags.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
"--auto-cast=none --model-type=transformer --tensorizer-options='--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-dge-dma --vectorize-strided-dma ' -O2 --internal-num-neuroncores-per-sengine=1 --logfile=/tmp/nxd_model/context_encoding_model/_tp0_bk0/log-neuron-cc.txt"
|
neuronxcc-2.17.194.0+d312836f/MODULE_374002f5e545876dbed9+bfe5714b/model.done
ADDED
File without changes
|
neuronxcc-2.17.194.0+d312836f/MODULE_374002f5e545876dbed9+bfe5714b/model.hlo_module.pb
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:82480b63e69fde09f4e77537a22aed3dea465afb13ef04e814662744ab15993b
|
3 |
+
size 76491
|
neuronxcc-2.17.194.0+d312836f/MODULE_374002f5e545876dbed9+bfe5714b/model.neff
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dd9b7366fe903881c52a003d874f50af91d5c94c4323aab4803b0718ad2ce874
|
3 |
+
size 287744
|
neuronxcc-2.17.194.0+d312836f/MODULE_40a0e75a65ac51fdd01a+613edded/model.neff
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 144384
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:67a4245ec970153c19f88ea1e6c239644af48f531cd128eaf43d20a49b0a8a44
|
3 |
size 144384
|
neuronxcc-2.17.194.0+d312836f/MODULE_4617ff1c720a11b46bd3+165e9558/compile_flags.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
"--auto-cast=none --model-type=transformer --tensorizer-options='--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-dge-dma --vectorize-strided-dma ' -O2 --internal-num-neuroncores-per-sengine=1 --logfile=/tmp/nxd_model/token_generation_model/_tp0_bk0/log-neuron-cc.txt --enable-internal-neff-wrapper"
|
neuronxcc-2.17.194.0+d312836f/MODULE_4617ff1c720a11b46bd3+165e9558/model.done
ADDED
File without changes
|
neuronxcc-2.17.194.0+d312836f/MODULE_4617ff1c720a11b46bd3+165e9558/model.hlo_module.pb
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b3e52820e19bbf945d45b897bc3900f508886cdb8efdff25515450a7f8d02f8c
|
3 |
+
size 777821
|
neuronxcc-2.17.194.0+d312836f/MODULE_4617ff1c720a11b46bd3+165e9558/model.neff
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:72f54a4af65ac42af3f85054285f1db2841f87c479752d76d7a20dac051e03a6
|
3 |
+
size 5889024
|
neuronxcc-2.17.194.0+d312836f/MODULE_4617ff1c720a11b46bd3+165e9558/wrapped_neff.hlo
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:54ea37c17f95777faafab50142f5108b134aa732136033c29ffb86aec876bc8e
|
3 |
+
size 6026968
|
neuronxcc-2.17.194.0+d312836f/MODULE_48bb18150a194c32e820+bfe5714b/compile_flags.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
"--auto-cast=none --model-type=transformer --tensorizer-options='--enable-ccop-compute-overlap --cc-pipeline-tiling-factor=2 --vectorize-dge-dma --vectorize-strided-dma ' -O2 --internal-num-neuroncores-per-sengine=1 --logfile=/tmp/nxd_model/context_encoding_model/_tp0_bk0/log-neuron-cc.txt"
|