v75 (Galaxy S24 / SM8650, soc_model 57): Llama-3.1-Nemotron-Nano-8B 8-shard W8 bundle — export 8/8 (runtime needs >8GB RAM)

#1
by flying101 - opened
.gitattributes CHANGED
@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  v81/tokenizer.json filter=lfs diff=lfs merge=lfs -text
 
 
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  v81/tokenizer.json filter=lfs diff=lfs merge=lfs -text
37
+ v75/tokenizer.json filter=lfs diff=lfs merge=lfs -text
v75/nemotron-nano-8b.json ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "_comment": "Llama-3.1-Nemotron-Nano-8B-v1 (NVIDIA reasoning fine-tune of Llama-3.1-8B) on Hexagon v81 (SM8850 / soc_model 87). Standard dense Llama-3.1-8B (32 q heads -> materialized MHA + MAXCTX 512; llama3-scaled RoPE theta 500000 factor 8; untied lm-head). The ~7GB W8 decode exceeds the v81 cDSP per-context ceiling, so the 32 layers are split into 8 CHAINED decode parts (decode_parts=8, ~0.85GB W8 each) driven by the llama_generate_sharded host-op (decode-over-prompt seed, no batched prefill). Export gate: decode 8/8 match gold (forge). W8 device gate is greedy_tol.",
4
+ "model": {
5
+ "name": "nemotron-nano-8b",
6
+ "family": "llm",
7
+ "dsp_arch": "v75",
8
+ "tokenizer_pre": "llama3"
9
+ },
10
+ "params": {
11
+ "hidden": 4096,
12
+ "vocab": 128256,
13
+ "n_layers": 32,
14
+ "max_ctx": 512,
15
+ "kv_dim": 1024,
16
+ "head_dim": 128,
17
+ "rope_theta": 500000.0,
18
+ "eos_token_id": 128001
19
+ },
20
+ "artifacts": {
21
+ "contexts": {
22
+ "decode_p0": {
23
+ "bin": "nemotronnano8b_decode_p0_w8.bin"
24
+ },
25
+ "decode_p1": {
26
+ "bin": "nemotronnano8b_decode_p1_w8.bin"
27
+ },
28
+ "decode_p2": {
29
+ "bin": "nemotronnano8b_decode_p2_w8.bin"
30
+ },
31
+ "decode_p3": {
32
+ "bin": "nemotronnano8b_decode_p3_w8.bin"
33
+ },
34
+ "decode_p4": {
35
+ "bin": "nemotronnano8b_decode_p4_w8.bin"
36
+ },
37
+ "decode_p5": {
38
+ "bin": "nemotronnano8b_decode_p5_w8.bin"
39
+ },
40
+ "decode_p6": {
41
+ "bin": "nemotronnano8b_decode_p6_w8.bin"
42
+ },
43
+ "decode_p7": {
44
+ "bin": "nemotronnano8b_decode_p7_w8.bin"
45
+ },
46
+ "lmhead": {
47
+ "bin": "nemotronnano8b_lmhead_w8.bin"
48
+ }
49
+ },
50
+ "embed": "nemotronnano8b_embed_f16.bin",
51
+ "tokenizer": "tokenizer.json"
52
+ },
53
+ "plan": {
54
+ "steps": [
55
+ {
56
+ "host": "llama_generate_sharded",
57
+ "params": {
58
+ "decode_parts": "nemotronnano8b_decode_p0_w8,nemotronnano8b_decode_p1_w8,nemotronnano8b_decode_p2_w8,nemotronnano8b_decode_p3_w8,nemotronnano8b_decode_p4_w8,nemotronnano8b_decode_p5_w8,nemotronnano8b_decode_p6_w8,nemotronnano8b_decode_p7_w8",
59
+ "lmhead": "nemotronnano8b_lmhead_w8",
60
+ "max_new": 64,
61
+ "neg": -50000,
62
+ "rope_factor": 8.0
63
+ }
64
+ }
65
+ ]
66
+ }
67
+ }
v75/nemotronnano8b_decode_p0_w8.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7b66e1eef4bc1903942bf09cb4823eddd9df57d20ea25d59244c4fa8d70eed76
3
+ size 876392448
v75/nemotronnano8b_decode_p1_w8.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0b0a9aab17bbd5c27a433dc267011368279d6d14e1949c4c44af2d18e458e0df
3
+ size 876392448
v75/nemotronnano8b_decode_p2_w8.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ca9a95042ae0e5f259174faca2cf14bd47b8169bd96c1e1d3771659773c6016b
3
+ size 876392448
v75/nemotronnano8b_decode_p3_w8.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a582d7bd4dbe8907ebe3e763329d8d3da7c5b58ea9ea1cc42d3262a4d871684a
3
+ size 876392448
v75/nemotronnano8b_decode_p4_w8.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb1463fcbdb1cedd3c7b7d181fc20e3c7d28541d00a7ff8d6d90d5b4ec23d3a0
3
+ size 876392448
v75/nemotronnano8b_decode_p5_w8.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f5d1a6ea2a9fc086992b2e87f672c1d6bd8e3ae4cf5ec8078efcdeb69810bb5f
3
+ size 876392448
v75/nemotronnano8b_decode_p6_w8.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:543a315646fd0803230111d06f8de370bb5a6d7ab3efaf55b76fd4fb647852b8
3
+ size 876392448
v75/nemotronnano8b_decode_p7_w8.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ffbfd021e63a2bdd0cae5e904a47b19125c4a63f0fa3299d91ef7b13dbc2ac24
3
+ size 876400640
v75/nemotronnano8b_embed_f16.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8443236c0685bf39fb998183ecd98d14e296cb6fee1e1c7841ec89d2c15fcfac
3
+ size 1050673152
v75/nemotronnano8b_lmhead_w8.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:844a917ae15d026d9b418c04e9726a0619f90208aca7a5987d2b80841353cb8a
3
+ size 527147008
v75/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b9e4e7fb171f92fd137b777cc2714bf87d11576700a1dcd7a399e7bbe39537b
3
+ size 17209920