Hugging Face's logo

aws-neuron
/
optimum-neuron-cache

Model card Files
xet
 Community
679
dacorvo HF Staff commited on
Commit
dca0c68
·
verified ·
1 Parent(s): d68d493

Synchronizing local compiler cache.

Browse files
Files changed (13) hide show
  1. .gitattributes +3 -0
  2. neuronxcc-2.21.18209.0+043b1bf7/0_REGISTRY/0.3.1.dev7/llama/princeton-nlp/Sheared-LLaMA-1.3B/fe12dbc4daf448a8271b.json +57 -0
  3. neuronxcc-2.21.18209.0+043b1bf7/MODULE_71777dc7066a131328e9+a9d440f5/compile_flags.json +1 -0
  4. neuronxcc-2.21.18209.0+043b1bf7/MODULE_71777dc7066a131328e9+a9d440f5/model.done +0 -0
  5. neuronxcc-2.21.18209.0+043b1bf7/MODULE_71777dc7066a131328e9+a9d440f5/model.hlo_module.pb +3 -0
  6. neuronxcc-2.21.18209.0+043b1bf7/MODULE_71777dc7066a131328e9+a9d440f5/model.neff +3 -0
  7. neuronxcc-2.21.18209.0+043b1bf7/MODULE_71777dc7066a131328e9+a9d440f5/wrapped_neff.hlo +3 -0
  8. neuronxcc-2.21.18209.0+043b1bf7/MODULE_d7a6d942281cbe0d0222+ed72d204/compile_flags.json +1 -0
  9. neuronxcc-2.21.18209.0+043b1bf7/MODULE_d7a6d942281cbe0d0222+ed72d204/model.done +0 -0
  10. neuronxcc-2.21.18209.0+043b1bf7/MODULE_d7a6d942281cbe0d0222+ed72d204/model.hlo_module.pb +3 -0
  11. neuronxcc-2.21.18209.0+043b1bf7/MODULE_d7a6d942281cbe0d0222+ed72d204/model.neff +3 -0
  12. neuronxcc-2.21.18209.0+043b1bf7/MODULE_e5881ad1904fef704fb7+a9d440f5/model.neff +1 -1
  13. neuronxcc-2.21.18209.0+043b1bf7/MODULE_e5881ad1904fef704fb7+a9d440f5/wrapped_neff.hlo +1 -1
.gitattributes CHANGED
@@ -11365,3 +11365,6 @@ neuronxcc-2.21.18209.0+043b1bf7/MODULE_9791b71b3f133f16e3d7+a9d440f5/wrapped_nef
11365
  neuronxcc-2.21.18209.0+043b1bf7/MODULE_7e5bb4d4caee6ec450d1+a9d440f5/model.neff filter=lfs diff=lfs merge=lfs -text
11366
  neuronxcc-2.21.18209.0+043b1bf7/MODULE_7e5bb4d4caee6ec450d1+a9d440f5/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
11367
  neuronxcc-2.21.18209.0+043b1bf7/MODULE_e113d55f1f361e1644f3+ed72d204/model.neff filter=lfs diff=lfs merge=lfs -text
 
 
 
 
11365
  neuronxcc-2.21.18209.0+043b1bf7/MODULE_7e5bb4d4caee6ec450d1+a9d440f5/model.neff filter=lfs diff=lfs merge=lfs -text
11366
  neuronxcc-2.21.18209.0+043b1bf7/MODULE_7e5bb4d4caee6ec450d1+a9d440f5/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
11367
  neuronxcc-2.21.18209.0+043b1bf7/MODULE_e113d55f1f361e1644f3+ed72d204/model.neff filter=lfs diff=lfs merge=lfs -text
11368
+ neuronxcc-2.21.18209.0+043b1bf7/MODULE_71777dc7066a131328e9+a9d440f5/model.neff filter=lfs diff=lfs merge=lfs -text
11369
+ neuronxcc-2.21.18209.0+043b1bf7/MODULE_71777dc7066a131328e9+a9d440f5/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
11370
+ neuronxcc-2.21.18209.0+043b1bf7/MODULE_d7a6d942281cbe0d0222+ed72d204/model.neff filter=lfs diff=lfs merge=lfs -text
neuronxcc-2.21.18209.0+043b1bf7/0_REGISTRY/0.3.1.dev7/llama/princeton-nlp/Sheared-LLaMA-1.3B/fe12dbc4daf448a8271b.json ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_entry_class": "SingleModelCacheEntry",
3
+ "_model_id": "princeton-nlp/Sheared-LLaMA-1.3B",
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": 2048,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 5504,
15
+ "max_position_embeddings": 4096,
16
+ "mlp_bias": false,
17
+ "model_type": "llama",
18
+ "neuron": {
19
+ "_serialized_key": "NxDNeuronConfig",
20
+ "batch_size": 4,
21
+ "capacity_factor": null,
22
+ "checkpoint_id": "princeton-nlp/Sheared-LLaMA-1.3B",
23
+ "checkpoint_revision": "a4b76938edbf571ea7d7d9904861cbdca08809b4",
24
+ "continuous_batching": true,
25
+ "enable_bucketing": false,
26
+ "ep_degree": 1,
27
+ "fused_qkv": true,
28
+ "glu_mlp": true,
29
+ "local_ranks_size": 2,
30
+ "logical_nc_config": 1,
31
+ "max_batch_size": 4,
32
+ "max_context_length": 4096,
33
+ "max_topk": 256,
34
+ "n_active_tokens": 4096,
35
+ "neuronxcc_version": "2.21.18209.0+043b1bf7",
36
+ "on_device_sampling": true,
37
+ "optimum_neuron_version": "0.3.1.dev7",
38
+ "output_logits": false,
39
+ "pp_degree": 1,
40
+ "sequence_length": 4096,
41
+ "speculation_length": 0,
42
+ "start_rank_id": 0,
43
+ "target": null,
44
+ "torch_dtype": "float16",
45
+ "tp_degree": 2
46
+ },
47
+ "num_attention_heads": 16,
48
+ "num_hidden_layers": 24,
49
+ "num_key_value_heads": 16,
50
+ "pretraining_tp": 1,
51
+ "rms_norm_eps": 1e-05,
52
+ "rope_scaling": null,
53
+ "rope_theta": 10000.0,
54
+ "tie_word_embeddings": false,
55
+ "use_cache": true,
56
+ "vocab_size": 32000
57
+ }
neuronxcc-2.21.18209.0+043b1bf7/MODULE_71777dc7066a131328e9+a9d440f5/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_model/_tp0_bk0/log-neuron-cc.txt", "--enable-internal-neff-wrapper"]
neuronxcc-2.21.18209.0+043b1bf7/MODULE_71777dc7066a131328e9+a9d440f5/model.done ADDED
File without changes
neuronxcc-2.21.18209.0+043b1bf7/MODULE_71777dc7066a131328e9+a9d440f5/model.hlo_module.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e5d44ac9bf286994a0dea52378d7736f08e79f6c24ba55d4753e9c7d19b198bc
3
+ size 531869
neuronxcc-2.21.18209.0+043b1bf7/MODULE_71777dc7066a131328e9+a9d440f5/model.neff ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1b5dcb554b0e5e69a5de02a2c557e49c4efd75ee6b7683c9165a849e165348d7
3
+ size 4895744
neuronxcc-2.21.18209.0+043b1bf7/MODULE_71777dc7066a131328e9+a9d440f5/wrapped_neff.hlo ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b735227b8d7e6f103f374307e7fe08cf88c932229823397d6cba1dc2d63b5ce1
3
+ size 5006267
neuronxcc-2.21.18209.0+043b1bf7/MODULE_d7a6d942281cbe0d0222+ed72d204/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_model/_tp0_bk0/log-neuron-cc.txt"]
neuronxcc-2.21.18209.0+043b1bf7/MODULE_d7a6d942281cbe0d0222+ed72d204/model.done ADDED
File without changes
neuronxcc-2.21.18209.0+043b1bf7/MODULE_d7a6d942281cbe0d0222+ed72d204/model.hlo_module.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9b2620fe32e93840aa6b41c5f07f7d86ee874d7c0ca5a225620248fd56a8bc4c
3
+ size 706537
neuronxcc-2.21.18209.0+043b1bf7/MODULE_d7a6d942281cbe0d0222+ed72d204/model.neff ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:48d92a2b316d98a36063659d077cf7811be7f1c8d4af7299daaeadd57eb36237
3
+ size 42937344
neuronxcc-2.21.18209.0+043b1bf7/MODULE_e5881ad1904fef704fb7+a9d440f5/model.neff CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5232114001bd4441991560bb14c60dc448174296ba57a9bf540da71f1c229c16
3
  size 6738944
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:863d71a8785735462742c9cfad7e18bd559721d9dcc9b03eb02ec51d488f666e
3
  size 6738944
neuronxcc-2.21.18209.0+043b1bf7/MODULE_e5881ad1904fef704fb7+a9d440f5/wrapped_neff.hlo CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e75fef7f30ab88b32f259fa4b5db1332cd4eb5883d94dc3de9580d1c4e7ef409
3
  size 6886551
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e7ef6c50222c9efd5ba6cd79fc8a2a3e85cdc9b816e9b7e65de7d3e9791ae51f
3
  size 6886551