soyrsoyr commited on
Commit
4217378
·
verified ·
1 Parent(s): 02d82af

Publish tested PR 3225 tiny MTP fixture

Browse files

Toy-trained checkpoint and validation evidence for llm-compressor 2d52420. Cards document architecture-specific limitations.

.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip 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
 
 
33
  *.zip 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
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,79 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ tags:
5
+ - llm-compressor
6
+ - tiny-model
7
+ - mtp
8
+ - pr-3225
9
+ - test-fixture
10
+ ---
11
+
12
+ # Inkling-0.92B-MTP-PR3225
13
+
14
+ Toy-trained BF16 source for [LLM Compressor PR #3225](https://github.com/vllm-project/llm-compressor/pull/3225). This is a small architecture and checkpoint test fixture, not a production language model. Its backbone was initialized randomly and trained on the repository tiny-model skill's toy text corpus; no pretrained base-model weights were used.
15
+
16
+ ## Model and provenance
17
+
18
+ - Original architecture/tokenizer: [thinkingmachines/Inkling](https://huggingface.co/thinkingmachines/Inkling), revision `828496eeae4c243ff1a22f7f28ff83694f2f7bc9`.
19
+ - Transformers class: `InklingForConditionalGeneration`.
20
+ - Total source parameters including synthetic MTP: **915,998,469** (0.916B).
21
+ - Backbone parameters: **830,199,043**; MTP parameters: **85,799,426**.
22
+ - Estimated active backbone parameters: **783,013,123**. This routing estimate includes embeddings/dense components, excludes MTP, and is not a FLOP measurement.
23
+ - Precision: BF16.
24
+ - Checkpoint: 4 indexed safetensors shards; 109 indexed tensors. Tokenizer and processor assets are included.
25
+
26
+ ## Configuration
27
+
28
+ | Field | Original | Tiny |
29
+ | --- | --- | --- |
30
+ | `num_hidden_layers` | 66 | 3 |
31
+ | `hidden_size` | 6144 | 1536 |
32
+ | `intermediate_size` | 3072 | 6144 |
33
+ | `moe_intermediate_size` | not explicitly set | 1024 |
34
+ | `n_routed_experts` | 256 | 16 |
35
+ | `num_experts_per_tok` | 6 | 6 |
36
+ | `num_attention_heads` | 64 | 16 |
37
+ | `num_key_value_heads` | 8 | 8 |
38
+ | `num_mtp_layers` | not explicitly set | 2 |
39
+
40
+ The saved `config.json` is authoritative. Depth is reduced; the limitations below apply.
41
+ Inkling retains sliding and full attention, dense and sparse MLPs, and its multimodal branches. Only text was trained/tested; image/audio quality was not evaluated.
42
+
43
+ ## Validation and scope
44
+
45
+ The full multimodal model's text-only training and backbone reload passed; its nonuniform attention/MLP layout is retained. Single-process pipeline loading passed. The two-rank pipeline failed during fused-MoE/meta tensor handling, and vLLM inspection encountered the environment's NumPy/Numba compatibility error. No architecture or loader override was applied. MFPTQ, quantized serving, and MTP execution have not been validated for this fixture.
46
+
47
+ The reloaded **BF16 backbone** toy perplexity was **1.570810** on the same small corpus used for training. This demonstrates learning/reload integrity, not generalization or benchmark quality.
48
+
49
+ MTP projections are synthetic initializations and decoder weights were copied from trained backbone blocks. **MTP was not separately trained**, so these artifacts do not establish draft acceptance quality or inference acceleration.
50
+
51
+ The tested model-based environment used **Transformers 5.17.0**, Torch 2.14.0+cu130, LLM Compressor [`2d52420`](https://github.com/vllm-project/llm-compressor/commit/2d5242056a8a00028686a303771ffc8e2fe27d07), and compressed-tensors [`e69c8dc`](https://github.com/vllm-project/compressed-tensors/commit/e69c8dc58aa152e5f5e36e85800ec1d2e7de5271). Serving smoke checks used vLLM 0.30.0. GLM/DeepSeek model-based MTP loading failed on Transformers **5.15.0**; **5.16 was not tested**. Calibration-dependent MTP quantization remains follow-up work.
52
+
53
+ The model class/loader was not patched to make tests pass. Construction changes were configuration reductions and explicit synthetic checkpoint fixture creation. Structured results and configuration provenance are in [validation.json](validation.json); [artifact-manifest.json](artifact-manifest.json) records checkpoint file hashes.
54
+
55
+ ## Backbone loading
56
+
57
+ With the tested Transformers environment (and compressed-tensors for FP8), this loads the backbone. MTP is opt-in; ordinary Transformers backbone generation does not execute it.
58
+
59
+ ```python
60
+ import torch
61
+ from transformers import InklingForConditionalGeneration, AutoTokenizer
62
+
63
+ model_id = "inference-optimization/Inkling-0.92B-MTP-PR3225"
64
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
65
+ model = InklingForConditionalGeneration.from_pretrained(
66
+ model_id, dtype=torch.bfloat16, attn_implementation="eager",
67
+ )
68
+ inputs = tokenizer("The capital of France is", return_tensors="pt")
69
+ output = model.generate(**inputs, max_new_tokens=16, do_sample=False)
70
+ print(tokenizer.decode(output[0], skip_special_tokens=True))
71
+ ```
72
+
73
+ ## Training
74
+
75
+ Random seed 3225; AdamW with learning rate 0.0004 and weight decay 0.01; batch size 2; text truncated to 160 tokens. Training stopped after three consecutive checks at toy perplexity <=3. The corpus follows the [repository tiny-model workflow](https://github.com/vllm-project/llm-compressor/tree/2d5242056a8a00028686a303771ffc8e2fe27d07/.agents/skills/create-tiny-model), with one additional short saying. These are the trained/reloaded artifacts used in testing.
76
+
77
+ ## License and attribution
78
+
79
+ Upstream architecture/tokenizer provenance and license: [thinkingmachines/Inkling](https://huggingface.co/thinkingmachines/Inkling/tree/828496eeae4c243ff1a22f7f28ff83694f2f7bc9).
artifact-manifest.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "repo_id": "inference-optimization/Inkling-0.92B-MTP-PR3225",
3
+ "files": {
4
+ "README.md": {
5
+ "size": 5061,
6
+ "sha256": "0e359db0aa89946ba67206a687d863822b260ced504331b5643ef516650acb11"
7
+ },
8
+ "chat_template.jinja": {
9
+ "size": 6294,
10
+ "sha256": "0aa1aa0c729d90176dcaa00c440c8faffca2957ffb2cc4b79456ee6d02bcf43b"
11
+ },
12
+ "config.json": {
13
+ "size": 2969,
14
+ "sha256": "17e1dad01f6c9770b10b94f7b60175327280c1bb7b71a49f43c1b187aba4ab36"
15
+ },
16
+ "generation_config.json": {
17
+ "size": 179,
18
+ "sha256": "122edeeea81ea41601f463a5f6fbe56a1d0e46b1ab2d6e1be2396785647ba4b5"
19
+ },
20
+ "model-00001-of-00003.safetensors": {
21
+ "size": 617545864,
22
+ "sha256": "9d916d3ca74f99a0b2f8bbc2f6e2322f8cc80a42b8a2c0e57c670220e57b2a77"
23
+ },
24
+ "model-00002-of-00003.safetensors": {
25
+ "size": 617545864,
26
+ "sha256": "2a067909dfe308248e1cf161128b86b30ff495833ad6fb8a5c4394d4996f0237"
27
+ },
28
+ "model-00003-of-00003.safetensors": {
29
+ "size": 425314046,
30
+ "sha256": "b8ef053454619d9d8cf42d79bbfa31c6430770e3c54224b32776ed14799d3d23"
31
+ },
32
+ "model-mtp.safetensors": {
33
+ "size": 171603828,
34
+ "sha256": "1c3b2574db8171c2d437d57b77e3fd164c8c1bea29f1e3c2938d6ab171ea1265"
35
+ },
36
+ "model.safetensors.index.json": {
37
+ "size": 8968,
38
+ "sha256": "9691ccd63bae634ecf34fcfcd0ac4418fad98532f3aa4f2d33d41e6feb8fa066"
39
+ },
40
+ "processor_config.json": {
41
+ "size": 1160,
42
+ "sha256": "3e10ba0f5b06b197bf194a2902ca108e0ffa0b895d14c40359f60e6a8964b752"
43
+ },
44
+ "tokenizer.json": {
45
+ "size": 27875796,
46
+ "sha256": "c470271fb7649eef922e660e757b5f076e2a94adc20df7fad7df85c46bc48eac"
47
+ },
48
+ "tokenizer_config.json": {
49
+ "size": 2011,
50
+ "sha256": "a5fac6a9ccffb5c681e2877cc749b612a8001dddcd65badaddeb00cadd1a750d"
51
+ },
52
+ "validation.json": {
53
+ "size": 15043,
54
+ "sha256": "ece68c2894ff25ffbe3450cf0ad9ca193f22196d9d9f10f3112ff008e6ee2eb7"
55
+ }
56
+ },
57
+ "weight_and_config_changes_during_publication": false
58
+ }
chat_template.jinja ADDED
@@ -0,0 +1,129 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- set effort_map = {"none": 0.0, "minimal": 0.1, "low": 0.2, "medium": 0.7, "high": 0.9, "max": 0.99} -%}
2
+ {%- set role_token = {"user": "<|message_user|>", "assistant": "<|message_model|>", "system": "<|message_system|>", "tool": "<|message_tool|>"} -%}
3
+
4
+ {%- macro emit_thinking_effort() -%}
5
+ {%- set eff = reasoning_effort if reasoning_effort is defined and reasoning_effort is not none else 0.9 -%}
6
+ {%- if eff is string -%}
7
+ {%- set key = eff | trim -%}
8
+ {%- if key not in effort_map -%}
9
+ {{- raise_exception("Unknown reasoning_effort: " ~ eff) -}}
10
+ {%- endif -%}
11
+ {%- set num = effort_map[key] -%}
12
+ {%- else -%}
13
+ {%- set num = eff | float -%}
14
+ {%- endif -%}
15
+ {%- if num < 0.0 or num > 0.99 -%}
16
+ {{- raise_exception("reasoning_effort must be in [0.0, 0.99]") -}}
17
+ {%- endif -%}
18
+ {{- "<|message_system|><|content_text|>Thinking effort level: " -}}
19
+ {%- if num == 0.0 -%}0{%- else -%}{{ num }}{%- endif -%}
20
+ {{- "<|end_message|>" -}}
21
+ {%- endmacro -%}
22
+
23
+ {%- if tools -%}
24
+ {%- set tool_state = namespace(specs=[]) -%}
25
+ {%- for tool in tools -%}
26
+ {%- set fn = tool.function if tool.function is defined else tool -%}
27
+ {%- set spec = {
28
+ "description": (fn.description if fn.description is defined and fn.description else ""),
29
+ "name": fn.name,
30
+ "parameters": (fn.parameters if fn.parameters is defined and fn.parameters else {}),
31
+ "type": (tool.type if tool.type is defined and tool.type else "function"),
32
+ } -%}
33
+ {%- set tool_state.specs = tool_state.specs + [spec] -%}
34
+ {%- endfor -%}
35
+ {{- "<|message_system|>tool_declare<|content_xml|>" -}}
36
+ {{- tool_state.specs | tojson(sort_keys=true, separators=(",", ":")) -}}
37
+ {{- "<|end_message|>" -}}
38
+ {%- endif -%}
39
+
40
+ {%- set state = namespace(effort_emitted=false) -%}
41
+ {%- for message in messages -%}
42
+ {%- if message.role not in role_token -%}
43
+ {{- raise_exception("Unknown message role: " ~ message.role) -}}
44
+ {%- endif -%}
45
+ {%- if not state.effort_emitted and message.role != "system" -%}
46
+ {{- emit_thinking_effort() -}}
47
+ {%- set state.effort_emitted = true -%}
48
+ {%- endif -%}
49
+
50
+ {%- set rtok = role_token[message.role] -%}
51
+
52
+ {%- if message.role == "tool" -%}
53
+ {%- set tool_name_state = namespace(name="") -%}
54
+ {%- if message.name is defined and message.name -%}
55
+ {%- set tool_name_state.name = message.name -%}
56
+ {%- elif message.tool_call_id is defined and message.tool_call_id -%}
57
+ {%- for prev in messages -%}
58
+ {%- if prev.role == "assistant" and prev.tool_calls -%}
59
+ {%- for tc in prev.tool_calls -%}
60
+ {%- if tc.id is defined and tc.id == message.tool_call_id and tc.function.name is defined -%}
61
+ {%- set tool_name_state.name = tc.function.name -%}
62
+ {%- endif -%}
63
+ {%- endfor -%}
64
+ {%- endif -%}
65
+ {%- endfor -%}
66
+ {%- endif -%}
67
+ {{- rtok -}}
68
+ {%- if tool_name_state.name -%}{{- tool_name_state.name -}}{%- endif -%}
69
+ {{- "<|content_text|>" -}}
70
+ {%- if message.content is string -%}{{- message.content -}}{%- endif -%}
71
+ {{- "<|end_message|>" -}}
72
+
73
+ {%- else -%}
74
+ {%- if message.role == "assistant" and message.reasoning_content is defined and message.reasoning_content -%}
75
+ {{- "<|message_model|><|content_thinking|>" ~ message.reasoning_content ~ "<|end_message|>" -}}
76
+ {%- endif -%}
77
+
78
+ {%- if message.content is string -%}
79
+ {{- rtok ~ "<|content_text|>" ~ message.content ~ "<|end_message|>" -}}
80
+ {%- elif message.content -%}
81
+ {%- for part in message.content -%}
82
+ {%- if part is string -%}
83
+ {{- rtok ~ "<|content_text|>" ~ part ~ "<|end_message|>" -}}
84
+ {%- elif part.type is not defined or part.type in ("text", "input_text") -%}
85
+ {%- set text_part = (part.text if part.text is defined and part.text is string else "") -%}
86
+ {{- rtok ~ "<|content_text|>" ~ text_part ~ "<|end_message|>" -}}
87
+ {%- elif part.type in ("image", "input_image", "image_url") -%}
88
+ {{- rtok ~ "<|content_image|><|unused_200054|><|end_message|>" -}}
89
+ {%- elif part.type in ("audio", "input_audio", "audio_url") -%}
90
+ {{- rtok ~ "<|content_audio_input|><|unused_200053|><|audio_end|><|end_message|>" -}}
91
+ {%- else -%}
92
+ {{- raise_exception("Unsupported content part type: " ~ part.type) -}}
93
+ {%- endif -%}
94
+ {%- endfor -%}
95
+ {%- endif -%}
96
+
97
+ {%- if message.role == "assistant" and message.tool_calls -%}
98
+ {%- for tc in message.tool_calls -%}
99
+ {%- set fn = tc.function -%}
100
+ {%- if fn.name is not defined or fn.name is not string -%}
101
+ {{- raise_exception("tool call function name must be a string") -}}
102
+ {%- endif -%}
103
+ {%- set args = fn.arguments if fn.arguments is defined and fn.arguments else {} -%}
104
+ {%- if args is string -%}
105
+ {{- raise_exception("tool call arguments must be a parsed object, not a JSON string; canonicalize upstream") -}}
106
+ {%- endif -%}
107
+ {%- if args is not mapping -%}
108
+ {{- raise_exception("tool call arguments must be an object") -}}
109
+ {%- endif -%}
110
+ {{- "<|message_model|>" ~ fn.name ~ "<|content_invoke_tool_json|>" -}}
111
+ {{- '{"name":' ~ (fn.name | tojson(sort_keys=true, separators=(",", ":"))) ~ ',"args":' -}}
112
+ {{- (args | tojson(sort_keys=true, separators=(",", ":"))) -}}
113
+ {{- "}<|end_message|>" -}}
114
+ {%- endfor -%}
115
+ {%- endif -%}
116
+
117
+ {%- if message.role == "assistant" -%}
118
+ {{- "<|content_model_end_sampling|>" -}}
119
+ {%- endif -%}
120
+ {%- endif -%}
121
+ {%- endfor -%}
122
+
123
+ {%- if not state.effort_emitted -%}
124
+ {{- emit_thinking_effort() -}}
125
+ {%- endif -%}
126
+
127
+ {%- if add_generation_prompt -%}
128
+ {{- "<|message_model|>" -}}
129
+ {%- endif -%}
config.json ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "InklingForConditionalGeneration"
4
+ ],
5
+ "audio_bos_token_id": 200020,
6
+ "audio_config": {
7
+ "audio_mode": "dmel",
8
+ "bias": false,
9
+ "decoder_dmodel": 6144,
10
+ "dmel_max_value": 2.0,
11
+ "dmel_min_value": -7.0,
12
+ "dtype": "bfloat16",
13
+ "initializer_range": 0.02,
14
+ "mel_vocab_size": 16,
15
+ "model_type": "inkling_audio",
16
+ "n_mel_bins": 80,
17
+ "rms_norm_eps": 1e-06,
18
+ "text_hidden_size": 1536,
19
+ "use_audio_norm": true
20
+ },
21
+ "audio_token_id": 200053,
22
+ "dtype": "bfloat16",
23
+ "eos_token_id": 200006,
24
+ "image_bos_token_id": 200005,
25
+ "image_token_id": 200054,
26
+ "model_type": "inkling_mm_model",
27
+ "mtp_config": {
28
+ "chain_hidden_post_norm": false,
29
+ "local_layer_ids": [
30
+ 0
31
+ ],
32
+ "num_nextn_predict_layers": 2
33
+ },
34
+ "pad_token_id": 200006,
35
+ "text_config": {
36
+ "attention_dropout": 0.0,
37
+ "bos_token_id": 1,
38
+ "chain_hidden_post_norm": false,
39
+ "conv_kernel_size": 4,
40
+ "d_rel": 16,
41
+ "dtype": "bfloat16",
42
+ "eos_token_id": 2,
43
+ "final_logit_softcapping": null,
44
+ "gate_activation": "sigmoid",
45
+ "head_dim": 128,
46
+ "hidden_act": "silu",
47
+ "hidden_size": 1536,
48
+ "initializer_range": 0.02,
49
+ "intermediate_size": 6144,
50
+ "layer_types": [
51
+ "hybrid_sliding",
52
+ "hybrid_sliding",
53
+ "hybrid"
54
+ ],
55
+ "local_layer_ids": [
56
+ 0,
57
+ 1
58
+ ],
59
+ "log_scaling_alpha": 0.1,
60
+ "log_scaling_n_floor": 128000,
61
+ "logits_mup_width_multiplier": 6.0,
62
+ "max_position_embeddings": 1048576,
63
+ "mlp_layer_types": [
64
+ "dense",
65
+ "dense",
66
+ "sparse"
67
+ ],
68
+ "model_type": "inkling_text",
69
+ "moe_intermediate_size": 1024,
70
+ "mtp_hidden_states_first": true,
71
+ "mtp_local_layer_ids": [
72
+ 0
73
+ ],
74
+ "n_routed_experts": 16,
75
+ "n_shared_experts": 2,
76
+ "norm_after_topk": true,
77
+ "num_attention_heads": 16,
78
+ "num_experts_per_tok": 6,
79
+ "num_hidden_layers": 3,
80
+ "num_key_value_heads": 8,
81
+ "num_mtp_layers": 2,
82
+ "number_of_conv_states": 4,
83
+ "o_bias": false,
84
+ "pad_token_id": null,
85
+ "q_bias": false,
86
+ "rel_extent": 1024,
87
+ "rms_norm_eps": 1e-06,
88
+ "rms_norm_eps_moe_gate": 1e-06,
89
+ "route_scale": 8.0,
90
+ "shared_expert_sink": true,
91
+ "sliding_window_size": 512,
92
+ "swa_head_dim": 128,
93
+ "swa_num_attention_heads": 16,
94
+ "swa_num_key_value_heads": 8,
95
+ "unpadded_vocab_size": 200058,
96
+ "use_embed_norm": true,
97
+ "use_gate_bias": true,
98
+ "use_global_scale": true,
99
+ "use_sconv": true,
100
+ "vocab_size": 201024
101
+ },
102
+ "transformers_version": "5.17.0",
103
+ "vision_config": {
104
+ "decoder_dmodel": 6144,
105
+ "dtype": "bfloat16",
106
+ "hidden_size": 1024,
107
+ "initializer_range": 0.02,
108
+ "model_type": "inkling_vision",
109
+ "n_channels": 3,
110
+ "n_layers": 4,
111
+ "num_attention_heads": 16,
112
+ "num_channels": 3,
113
+ "patch_size": 40,
114
+ "rms_norm_eps": 1e-06,
115
+ "temporal_patch_size": 2,
116
+ "text_hidden_size": 1536,
117
+ "use_vision_norm": true,
118
+ "vision_encoder_type": "hmlp"
119
+ }
120
+ }
generation_config.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 1,
4
+ "eos_token_id": 200006,
5
+ "output_attentions": false,
6
+ "output_hidden_states": false,
7
+ "transformers_version": "5.17.0"
8
+ }
model-00001-of-00003.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9d916d3ca74f99a0b2f8bbc2f6e2322f8cc80a42b8a2c0e57c670220e57b2a77
3
+ size 617545864
model-00002-of-00003.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a067909dfe308248e1cf161128b86b30ff495833ad6fb8a5c4394d4996f0237
3
+ size 617545864
model-00003-of-00003.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b8ef053454619d9d8cf42d79bbfa31c6430770e3c54224b32776ed14799d3d23
3
+ size 425314046
model-mtp.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1c3b2574db8171c2d437d57b77e3fd164c8c1bea29f1e3c2938d6ab171ea1265
3
+ size 171603828
model.safetensors.index.json ADDED
@@ -0,0 +1,116 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_size": 1832009602
4
+ },
5
+ "weight_map": {
6
+ "model.audio.encoder.weight": "model-00003-of-00003.safetensors",
7
+ "model.audio.final_norm.weight": "model-00003-of-00003.safetensors",
8
+ "model.llm.embed.weight": "model-00002-of-00003.safetensors",
9
+ "model.llm.embed_norm.weight": "model-00003-of-00003.safetensors",
10
+ "model.llm.layers.0.attn.k_norm.weight": "model-00003-of-00003.safetensors",
11
+ "model.llm.layers.0.attn.k_sconv.weight": "model-00003-of-00003.safetensors",
12
+ "model.llm.layers.0.attn.q_norm.weight": "model-00003-of-00003.safetensors",
13
+ "model.llm.layers.0.attn.rel_logits_proj.proj": "model-00003-of-00003.safetensors",
14
+ "model.llm.layers.0.attn.v_sconv.weight": "model-00003-of-00003.safetensors",
15
+ "model.llm.layers.0.attn.wk_dv.weight": "model-00003-of-00003.safetensors",
16
+ "model.llm.layers.0.attn.wo_ud.weight": "model-00003-of-00003.safetensors",
17
+ "model.llm.layers.0.attn.wq_du.weight": "model-00003-of-00003.safetensors",
18
+ "model.llm.layers.0.attn.wr_du.weight": "model-00003-of-00003.safetensors",
19
+ "model.llm.layers.0.attn.wv_dv.weight": "model-00003-of-00003.safetensors",
20
+ "model.llm.layers.0.attn_norm.weight": "model-00003-of-00003.safetensors",
21
+ "model.llm.layers.0.attn_sconv.weight": "model-00003-of-00003.safetensors",
22
+ "model.llm.layers.0.mlp.global_scale": "model-00003-of-00003.safetensors",
23
+ "model.llm.layers.0.mlp.w13_dn.weight": "model-00003-of-00003.safetensors",
24
+ "model.llm.layers.0.mlp.w2_md.weight": "model-00003-of-00003.safetensors",
25
+ "model.llm.layers.0.mlp_norm.weight": "model-00003-of-00003.safetensors",
26
+ "model.llm.layers.0.mlp_sconv.weight": "model-00003-of-00003.safetensors",
27
+ "model.llm.layers.1.attn.k_norm.weight": "model-00003-of-00003.safetensors",
28
+ "model.llm.layers.1.attn.k_sconv.weight": "model-00003-of-00003.safetensors",
29
+ "model.llm.layers.1.attn.q_norm.weight": "model-00003-of-00003.safetensors",
30
+ "model.llm.layers.1.attn.rel_logits_proj.proj": "model-00003-of-00003.safetensors",
31
+ "model.llm.layers.1.attn.v_sconv.weight": "model-00003-of-00003.safetensors",
32
+ "model.llm.layers.1.attn.wk_dv.weight": "model-00003-of-00003.safetensors",
33
+ "model.llm.layers.1.attn.wo_ud.weight": "model-00003-of-00003.safetensors",
34
+ "model.llm.layers.1.attn.wq_du.weight": "model-00003-of-00003.safetensors",
35
+ "model.llm.layers.1.attn.wr_du.weight": "model-00003-of-00003.safetensors",
36
+ "model.llm.layers.1.attn.wv_dv.weight": "model-00003-of-00003.safetensors",
37
+ "model.llm.layers.1.attn_norm.weight": "model-00003-of-00003.safetensors",
38
+ "model.llm.layers.1.attn_sconv.weight": "model-00003-of-00003.safetensors",
39
+ "model.llm.layers.1.mlp.global_scale": "model-00003-of-00003.safetensors",
40
+ "model.llm.layers.1.mlp.w13_dn.weight": "model-00003-of-00003.safetensors",
41
+ "model.llm.layers.1.mlp.w2_md.weight": "model-00003-of-00003.safetensors",
42
+ "model.llm.layers.1.mlp_norm.weight": "model-00003-of-00003.safetensors",
43
+ "model.llm.layers.1.mlp_sconv.weight": "model-00003-of-00003.safetensors",
44
+ "model.llm.layers.2.attn.k_norm.weight": "model-00003-of-00003.safetensors",
45
+ "model.llm.layers.2.attn.k_sconv.weight": "model-00003-of-00003.safetensors",
46
+ "model.llm.layers.2.attn.q_norm.weight": "model-00003-of-00003.safetensors",
47
+ "model.llm.layers.2.attn.rel_logits_proj.proj": "model-00003-of-00003.safetensors",
48
+ "model.llm.layers.2.attn.v_sconv.weight": "model-00003-of-00003.safetensors",
49
+ "model.llm.layers.2.attn.wk_dv.weight": "model-00003-of-00003.safetensors",
50
+ "model.llm.layers.2.attn.wo_ud.weight": "model-00003-of-00003.safetensors",
51
+ "model.llm.layers.2.attn.wq_du.weight": "model-00003-of-00003.safetensors",
52
+ "model.llm.layers.2.attn.wr_du.weight": "model-00003-of-00003.safetensors",
53
+ "model.llm.layers.2.attn.wv_dv.weight": "model-00003-of-00003.safetensors",
54
+ "model.llm.layers.2.attn_norm.weight": "model-00003-of-00003.safetensors",
55
+ "model.llm.layers.2.attn_sconv.weight": "model-00003-of-00003.safetensors",
56
+ "model.llm.layers.2.mlp.experts.w13_weight": "model-00003-of-00003.safetensors",
57
+ "model.llm.layers.2.mlp.experts.w2_weight": "model-00003-of-00003.safetensors",
58
+ "model.llm.layers.2.mlp.gate.bias": "model-00003-of-00003.safetensors",
59
+ "model.llm.layers.2.mlp.gate.global_scale": "model-00003-of-00003.safetensors",
60
+ "model.llm.layers.2.mlp.gate.weight": "model-00003-of-00003.safetensors",
61
+ "model.llm.layers.2.mlp.shared_experts.shared_w13_weight": "model-00003-of-00003.safetensors",
62
+ "model.llm.layers.2.mlp.shared_experts.shared_w2_weight": "model-00003-of-00003.safetensors",
63
+ "model.llm.layers.2.mlp_norm.weight": "model-00003-of-00003.safetensors",
64
+ "model.llm.layers.2.mlp_sconv.weight": "model-00003-of-00003.safetensors",
65
+ "model.llm.norm.weight": "model-00003-of-00003.safetensors",
66
+ "model.llm.unembed.weight": "model-00001-of-00003.safetensors",
67
+ "model.mtp.layers.0.embed_norm.weight": "model-mtp.safetensors",
68
+ "model.mtp.layers.0.hidden_norm.weight": "model-mtp.safetensors",
69
+ "model.mtp.layers.0.input_proj.weight": "model-mtp.safetensors",
70
+ "model.mtp.layers.0.transformer_block.attn.k_norm.weight": "model-mtp.safetensors",
71
+ "model.mtp.layers.0.transformer_block.attn.k_sconv.weight": "model-mtp.safetensors",
72
+ "model.mtp.layers.0.transformer_block.attn.q_norm.weight": "model-mtp.safetensors",
73
+ "model.mtp.layers.0.transformer_block.attn.rel_logits_proj.proj": "model-mtp.safetensors",
74
+ "model.mtp.layers.0.transformer_block.attn.v_sconv.weight": "model-mtp.safetensors",
75
+ "model.mtp.layers.0.transformer_block.attn.wk_dv.weight": "model-mtp.safetensors",
76
+ "model.mtp.layers.0.transformer_block.attn.wo_ud.weight": "model-mtp.safetensors",
77
+ "model.mtp.layers.0.transformer_block.attn.wq_du.weight": "model-mtp.safetensors",
78
+ "model.mtp.layers.0.transformer_block.attn.wr_du.weight": "model-mtp.safetensors",
79
+ "model.mtp.layers.0.transformer_block.attn.wv_dv.weight": "model-mtp.safetensors",
80
+ "model.mtp.layers.0.transformer_block.attn_norm.weight": "model-mtp.safetensors",
81
+ "model.mtp.layers.0.transformer_block.attn_sconv.weight": "model-mtp.safetensors",
82
+ "model.mtp.layers.0.transformer_block.mlp.global_scale": "model-mtp.safetensors",
83
+ "model.mtp.layers.0.transformer_block.mlp.w13_dn.weight": "model-mtp.safetensors",
84
+ "model.mtp.layers.0.transformer_block.mlp.w2_md.weight": "model-mtp.safetensors",
85
+ "model.mtp.layers.0.transformer_block.mlp_norm.weight": "model-mtp.safetensors",
86
+ "model.mtp.layers.0.transformer_block.mlp_sconv.weight": "model-mtp.safetensors",
87
+ "model.mtp.layers.1.embed_norm.weight": "model-mtp.safetensors",
88
+ "model.mtp.layers.1.hidden_norm.weight": "model-mtp.safetensors",
89
+ "model.mtp.layers.1.input_proj.weight": "model-mtp.safetensors",
90
+ "model.mtp.layers.1.transformer_block.attn.k_norm.weight": "model-mtp.safetensors",
91
+ "model.mtp.layers.1.transformer_block.attn.k_sconv.weight": "model-mtp.safetensors",
92
+ "model.mtp.layers.1.transformer_block.attn.q_norm.weight": "model-mtp.safetensors",
93
+ "model.mtp.layers.1.transformer_block.attn.rel_logits_proj.proj": "model-mtp.safetensors",
94
+ "model.mtp.layers.1.transformer_block.attn.v_sconv.weight": "model-mtp.safetensors",
95
+ "model.mtp.layers.1.transformer_block.attn.wk_dv.weight": "model-mtp.safetensors",
96
+ "model.mtp.layers.1.transformer_block.attn.wo_ud.weight": "model-mtp.safetensors",
97
+ "model.mtp.layers.1.transformer_block.attn.wq_du.weight": "model-mtp.safetensors",
98
+ "model.mtp.layers.1.transformer_block.attn.wr_du.weight": "model-mtp.safetensors",
99
+ "model.mtp.layers.1.transformer_block.attn.wv_dv.weight": "model-mtp.safetensors",
100
+ "model.mtp.layers.1.transformer_block.attn_norm.weight": "model-mtp.safetensors",
101
+ "model.mtp.layers.1.transformer_block.attn_sconv.weight": "model-mtp.safetensors",
102
+ "model.mtp.layers.1.transformer_block.mlp.global_scale": "model-mtp.safetensors",
103
+ "model.mtp.layers.1.transformer_block.mlp.w13_dn.weight": "model-mtp.safetensors",
104
+ "model.mtp.layers.1.transformer_block.mlp.w2_md.weight": "model-mtp.safetensors",
105
+ "model.mtp.layers.1.transformer_block.mlp_norm.weight": "model-mtp.safetensors",
106
+ "model.mtp.layers.1.transformer_block.mlp_sconv.weight": "model-mtp.safetensors",
107
+ "model.visual.final_norm.weight": "model-00003-of-00003.safetensors",
108
+ "model.visual.layers.linear_0.weight": "model-00003-of-00003.safetensors",
109
+ "model.visual.layers.linear_1.weight": "model-00003-of-00003.safetensors",
110
+ "model.visual.layers.linear_2.weight": "model-00003-of-00003.safetensors",
111
+ "model.visual.layers.linear_3.weight": "model-00003-of-00003.safetensors",
112
+ "model.visual.layers.norm_0.weight": "model-00003-of-00003.safetensors",
113
+ "model.visual.layers.norm_1.weight": "model-00003-of-00003.safetensors",
114
+ "model.visual.layers.norm_2.weight": "model-00003-of-00003.safetensors"
115
+ }
116
+ }
processor_config.json ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "audio_bos_token": "<|content_audio_input|>",
3
+ "audio_token": "<|unused_200053|>",
4
+ "dmel_max_value": 2.0,
5
+ "dmel_min_value": -7.0,
6
+ "feature_extractor": {
7
+ "audio_token_duration_s": 0.05,
8
+ "feature_extractor_type": "InklingFeatureExtractor",
9
+ "feature_size": 80,
10
+ "hop_length": 800,
11
+ "n_fft": 1600,
12
+ "padding_side": "right",
13
+ "padding_value": 0.0,
14
+ "return_attention_mask": true,
15
+ "sampling_rate": 16000,
16
+ "window_size": 1600,
17
+ "window_size_multiplier": 2.0
18
+ },
19
+ "image_bos_token": "<|content_image|>",
20
+ "image_processor": {
21
+ "do_convert_rgb": true,
22
+ "do_normalize": true,
23
+ "do_rescale": true,
24
+ "do_resize": true,
25
+ "image_mean": [
26
+ 0.48145466,
27
+ 0.4578275,
28
+ 0.40821073
29
+ ],
30
+ "image_processor_type": "InklingImageProcessor",
31
+ "image_std": [
32
+ 0.26862954,
33
+ 0.26130258,
34
+ 0.27577711
35
+ ],
36
+ "resample": 3,
37
+ "rescale_factor": 0.00392156862745098,
38
+ "rescale_image_max_upscaled_long_edge": 2048,
39
+ "size": {
40
+ "height": 40,
41
+ "width": 40
42
+ }
43
+ },
44
+ "image_token": "<|unused_200054|>",
45
+ "num_dmel_bins": 16,
46
+ "processor_class": "InklingProcessor"
47
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c470271fb7649eef922e660e757b5f076e2a94adc20df7fad7df85c46bc48eac
3
+ size 27875796
tokenizer_config.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "audio": "<|audio|>",
3
+ "audio_end": "<|audio_end|>",
4
+ "backend": "tokenizers",
5
+ "begin_of_text": "<|begin_of_text|>",
6
+ "clean_up_tokenization_spaces": false,
7
+ "content_audio_input": "<|content_audio_input|>",
8
+ "content_image": "<|content_image|>",
9
+ "content_invoke_tool_json": "<|content_invoke_tool_json|>",
10
+ "content_invoke_tool_text": "<|content_invoke_tool_text|>",
11
+ "content_model_end_sampling": "<|content_model_end_sampling|>",
12
+ "content_text": "<|content_text|>",
13
+ "content_thinking": "<|content_thinking|>",
14
+ "content_tool_error": "<|content_tool_error|>",
15
+ "content_xml": "<|content_xml|>",
16
+ "end_message": "<|end_message|>",
17
+ "endoftext": "<|endoftext|>",
18
+ "eos_token": "<|content_model_end_sampling|>",
19
+ "fix_mistral_regex": false,
20
+ "is_local": true,
21
+ "local_files_only": false,
22
+ "message_model": "<|message_model|>",
23
+ "message_system": "<|message_system|>",
24
+ "message_tool": "<|message_tool|>",
25
+ "message_user": "<|message_user|>",
26
+ "model_max_length": 1000000000000000019884624838656,
27
+ "model_specific_special_tokens": {
28
+ "audio": "<|audio|>",
29
+ "audio_end": "<|audio_end|>",
30
+ "begin_of_text": "<|begin_of_text|>",
31
+ "content_audio_input": "<|content_audio_input|>",
32
+ "content_image": "<|content_image|>",
33
+ "content_invoke_tool_json": "<|content_invoke_tool_json|>",
34
+ "content_invoke_tool_text": "<|content_invoke_tool_text|>",
35
+ "content_model_end_sampling": "<|content_model_end_sampling|>",
36
+ "content_text": "<|content_text|>",
37
+ "content_thinking": "<|content_thinking|>",
38
+ "content_tool_error": "<|content_tool_error|>",
39
+ "content_xml": "<|content_xml|>",
40
+ "end_message": "<|end_message|>",
41
+ "endoftext": "<|endoftext|>",
42
+ "message_model": "<|message_model|>",
43
+ "message_system": "<|message_system|>",
44
+ "message_tool": "<|message_tool|>",
45
+ "message_user": "<|message_user|>"
46
+ },
47
+ "pad_token": "<|content_model_end_sampling|>",
48
+ "processor_class": "InklingProcessor",
49
+ "tokenizer_class": "TokenizersBackend"
50
+ }
validation.json ADDED
@@ -0,0 +1,549 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "llm_compressor_commit": "2d5242056a8a00028686a303771ffc8e2fe27d07",
3
+ "compressed_tensors_commit": "e69c8dc58aa152e5f5e36e85800ec1d2e7de5271",
4
+ "transformers": "5.17.0",
5
+ "torch": "2.14.0+cu130",
6
+ "vllm": "0.30.0",
7
+ "base_architecture": "thinkingmachines/Inkling",
8
+ "base_revision": "828496eeae4c243ff1a22f7f28ff83694f2f7bc9",
9
+ "fixture": "inkling",
10
+ "variant": "source",
11
+ "parameter_counts": {
12
+ "architecture": "InklingForConditionalGeneration",
13
+ "layers_only_parameters": 62689820678,
14
+ "backbone_parameters": 830199043,
15
+ "mtp_parameters": 85799426,
16
+ "total_parameters": 915998469,
17
+ "changes": {
18
+ "dtype": {
19
+ "original": null,
20
+ "tiny": "bfloat16"
21
+ },
22
+ "text_config": {
23
+ "original": {
24
+ "architectures": null,
25
+ "output_hidden_states": false,
26
+ "return_dict": true,
27
+ "dtype": "bfloat16",
28
+ "chunk_size_feed_forward": 0,
29
+ "is_encoder_decoder": false,
30
+ "id2label": {
31
+ "0": "LABEL_0",
32
+ "1": "LABEL_1"
33
+ },
34
+ "label2id": {
35
+ "LABEL_0": 0,
36
+ "LABEL_1": 1
37
+ },
38
+ "problem_type": null,
39
+ "vocab_size": 201024,
40
+ "unpadded_vocab_size": 200058,
41
+ "hidden_size": 6144,
42
+ "num_hidden_layers": 66,
43
+ "num_attention_heads": 64,
44
+ "num_key_value_heads": 8,
45
+ "head_dim": 128,
46
+ "swa_num_attention_heads": 64,
47
+ "swa_num_key_value_heads": 16,
48
+ "swa_head_dim": 128,
49
+ "sliding_window_size": 512,
50
+ "d_rel": 16,
51
+ "rel_extent": 1024,
52
+ "log_scaling_n_floor": 128000,
53
+ "log_scaling_alpha": 0.1,
54
+ "local_layer_ids": [
55
+ 0,
56
+ 1,
57
+ 2,
58
+ 3,
59
+ 4,
60
+ 6,
61
+ 7,
62
+ 8,
63
+ 9,
64
+ 10,
65
+ 12,
66
+ 13,
67
+ 14,
68
+ 15,
69
+ 16,
70
+ 18,
71
+ 19,
72
+ 20,
73
+ 21,
74
+ 22,
75
+ 24,
76
+ 25,
77
+ 26,
78
+ 27,
79
+ 28,
80
+ 30,
81
+ 31,
82
+ 32,
83
+ 33,
84
+ 34,
85
+ 36,
86
+ 37,
87
+ 38,
88
+ 39,
89
+ 40,
90
+ 42,
91
+ 43,
92
+ 44,
93
+ 45,
94
+ 46,
95
+ 48,
96
+ 49,
97
+ 50,
98
+ 51,
99
+ 52,
100
+ 54,
101
+ 55,
102
+ 56,
103
+ 57,
104
+ 58,
105
+ 60,
106
+ 61,
107
+ 62,
108
+ 63,
109
+ 64
110
+ ],
111
+ "layer_types": [
112
+ "hybrid_sliding",
113
+ "hybrid_sliding",
114
+ "hybrid_sliding",
115
+ "hybrid_sliding",
116
+ "hybrid_sliding",
117
+ "hybrid",
118
+ "hybrid_sliding",
119
+ "hybrid_sliding",
120
+ "hybrid_sliding",
121
+ "hybrid_sliding",
122
+ "hybrid_sliding",
123
+ "hybrid",
124
+ "hybrid_sliding",
125
+ "hybrid_sliding",
126
+ "hybrid_sliding",
127
+ "hybrid_sliding",
128
+ "hybrid_sliding",
129
+ "hybrid",
130
+ "hybrid_sliding",
131
+ "hybrid_sliding",
132
+ "hybrid_sliding",
133
+ "hybrid_sliding",
134
+ "hybrid_sliding",
135
+ "hybrid",
136
+ "hybrid_sliding",
137
+ "hybrid_sliding",
138
+ "hybrid_sliding",
139
+ "hybrid_sliding",
140
+ "hybrid_sliding",
141
+ "hybrid",
142
+ "hybrid_sliding",
143
+ "hybrid_sliding",
144
+ "hybrid_sliding",
145
+ "hybrid_sliding",
146
+ "hybrid_sliding",
147
+ "hybrid",
148
+ "hybrid_sliding",
149
+ "hybrid_sliding",
150
+ "hybrid_sliding",
151
+ "hybrid_sliding",
152
+ "hybrid_sliding",
153
+ "hybrid",
154
+ "hybrid_sliding",
155
+ "hybrid_sliding",
156
+ "hybrid_sliding",
157
+ "hybrid_sliding",
158
+ "hybrid_sliding",
159
+ "hybrid",
160
+ "hybrid_sliding",
161
+ "hybrid_sliding",
162
+ "hybrid_sliding",
163
+ "hybrid_sliding",
164
+ "hybrid_sliding",
165
+ "hybrid",
166
+ "hybrid_sliding",
167
+ "hybrid_sliding",
168
+ "hybrid_sliding",
169
+ "hybrid_sliding",
170
+ "hybrid_sliding",
171
+ "hybrid",
172
+ "hybrid_sliding",
173
+ "hybrid_sliding",
174
+ "hybrid_sliding",
175
+ "hybrid_sliding",
176
+ "hybrid_sliding",
177
+ "hybrid"
178
+ ],
179
+ "max_position_embeddings": 1048576,
180
+ "rms_norm_eps": 1e-06,
181
+ "conv_kernel_size": 4,
182
+ "mlp_layer_types": [
183
+ "dense",
184
+ "dense",
185
+ "sparse",
186
+ "sparse",
187
+ "sparse",
188
+ "sparse",
189
+ "sparse",
190
+ "sparse",
191
+ "sparse",
192
+ "sparse",
193
+ "sparse",
194
+ "sparse",
195
+ "sparse",
196
+ "sparse",
197
+ "sparse",
198
+ "sparse",
199
+ "sparse",
200
+ "sparse",
201
+ "sparse",
202
+ "sparse",
203
+ "sparse",
204
+ "sparse",
205
+ "sparse",
206
+ "sparse",
207
+ "sparse",
208
+ "sparse",
209
+ "sparse",
210
+ "sparse",
211
+ "sparse",
212
+ "sparse",
213
+ "sparse",
214
+ "sparse",
215
+ "sparse",
216
+ "sparse",
217
+ "sparse",
218
+ "sparse",
219
+ "sparse",
220
+ "sparse",
221
+ "sparse",
222
+ "sparse",
223
+ "sparse",
224
+ "sparse",
225
+ "sparse",
226
+ "sparse",
227
+ "sparse",
228
+ "sparse",
229
+ "sparse",
230
+ "sparse",
231
+ "sparse",
232
+ "sparse",
233
+ "sparse",
234
+ "sparse",
235
+ "sparse",
236
+ "sparse",
237
+ "sparse",
238
+ "sparse",
239
+ "sparse",
240
+ "sparse",
241
+ "sparse",
242
+ "sparse",
243
+ "sparse",
244
+ "sparse",
245
+ "sparse",
246
+ "sparse",
247
+ "sparse",
248
+ "sparse"
249
+ ],
250
+ "intermediate_size": 24576,
251
+ "hidden_act": "silu",
252
+ "moe_intermediate_size": 3072,
253
+ "n_routed_experts": 256,
254
+ "num_experts_per_tok": 6,
255
+ "n_shared_experts": 2,
256
+ "shared_expert_sink": true,
257
+ "route_scale": 8.0,
258
+ "logits_mup_width_multiplier": 24.0,
259
+ "rms_norm_eps_moe_gate": 1e-06,
260
+ "attention_dropout": 0.0,
261
+ "initializer_range": 0.02,
262
+ "pad_token_id": null,
263
+ "bos_token_id": 1,
264
+ "eos_token_id": 2,
265
+ "num_mtp_layers": 8,
266
+ "chain_hidden_post_norm": false,
267
+ "mtp_hidden_states_first": true,
268
+ "mtp_local_layer_ids": [
269
+ 0,
270
+ 2,
271
+ 4,
272
+ 5,
273
+ 6,
274
+ 7
275
+ ],
276
+ "number_of_conv_states": 4,
277
+ "_name_or_path": "",
278
+ "q_bias": false,
279
+ "o_bias": false,
280
+ "use_embed_norm": true,
281
+ "use_sconv": true,
282
+ "final_logit_softcapping": null,
283
+ "use_gate_bias": true,
284
+ "gate_activation": "sigmoid",
285
+ "norm_after_topk": true,
286
+ "use_global_scale": true,
287
+ "model_type": "inkling_text",
288
+ "output_attentions": false
289
+ },
290
+ "tiny": {
291
+ "architectures": null,
292
+ "output_hidden_states": false,
293
+ "return_dict": true,
294
+ "dtype": "bfloat16",
295
+ "chunk_size_feed_forward": 0,
296
+ "is_encoder_decoder": false,
297
+ "id2label": {
298
+ "0": "LABEL_0",
299
+ "1": "LABEL_1"
300
+ },
301
+ "label2id": {
302
+ "LABEL_0": 0,
303
+ "LABEL_1": 1
304
+ },
305
+ "problem_type": null,
306
+ "vocab_size": 201024,
307
+ "unpadded_vocab_size": 200058,
308
+ "hidden_size": 1536,
309
+ "num_hidden_layers": 3,
310
+ "num_attention_heads": 16,
311
+ "num_key_value_heads": 8,
312
+ "head_dim": 128,
313
+ "swa_num_attention_heads": 16,
314
+ "swa_num_key_value_heads": 8,
315
+ "swa_head_dim": 128,
316
+ "sliding_window_size": 512,
317
+ "d_rel": 16,
318
+ "rel_extent": 1024,
319
+ "log_scaling_n_floor": 128000,
320
+ "log_scaling_alpha": 0.1,
321
+ "local_layer_ids": [
322
+ 0,
323
+ 1
324
+ ],
325
+ "layer_types": [
326
+ "hybrid_sliding",
327
+ "hybrid_sliding",
328
+ "hybrid"
329
+ ],
330
+ "max_position_embeddings": 1048576,
331
+ "rms_norm_eps": 1e-06,
332
+ "conv_kernel_size": 4,
333
+ "mlp_layer_types": [
334
+ "dense",
335
+ "dense",
336
+ "sparse"
337
+ ],
338
+ "intermediate_size": 6144,
339
+ "hidden_act": "silu",
340
+ "moe_intermediate_size": 1024,
341
+ "n_routed_experts": 16,
342
+ "num_experts_per_tok": 6,
343
+ "n_shared_experts": 2,
344
+ "shared_expert_sink": true,
345
+ "route_scale": 8.0,
346
+ "logits_mup_width_multiplier": 6.0,
347
+ "rms_norm_eps_moe_gate": 1e-06,
348
+ "attention_dropout": 0.0,
349
+ "initializer_range": 0.02,
350
+ "pad_token_id": null,
351
+ "bos_token_id": 1,
352
+ "eos_token_id": 2,
353
+ "num_mtp_layers": 2,
354
+ "chain_hidden_post_norm": false,
355
+ "mtp_hidden_states_first": true,
356
+ "mtp_local_layer_ids": [
357
+ 0
358
+ ],
359
+ "number_of_conv_states": 4,
360
+ "_name_or_path": "",
361
+ "q_bias": false,
362
+ "o_bias": false,
363
+ "use_embed_norm": true,
364
+ "use_sconv": true,
365
+ "final_logit_softcapping": null,
366
+ "use_gate_bias": true,
367
+ "gate_activation": "sigmoid",
368
+ "norm_after_topk": true,
369
+ "use_global_scale": true,
370
+ "model_type": "inkling_text",
371
+ "output_attentions": false
372
+ }
373
+ },
374
+ "audio_config": {
375
+ "original": {
376
+ "architectures": null,
377
+ "output_hidden_states": false,
378
+ "return_dict": true,
379
+ "dtype": null,
380
+ "chunk_size_feed_forward": 0,
381
+ "is_encoder_decoder": false,
382
+ "id2label": {
383
+ "0": "LABEL_0",
384
+ "1": "LABEL_1"
385
+ },
386
+ "label2id": {
387
+ "LABEL_0": 0,
388
+ "LABEL_1": 1
389
+ },
390
+ "problem_type": null,
391
+ "n_mel_bins": 80,
392
+ "mel_vocab_size": 16,
393
+ "text_hidden_size": 6144,
394
+ "rms_norm_eps": 1e-06,
395
+ "initializer_range": 0.02,
396
+ "_name_or_path": "",
397
+ "decoder_dmodel": 6144,
398
+ "bias": false,
399
+ "dmel_min_value": -7.0,
400
+ "dmel_max_value": 2.0,
401
+ "use_audio_norm": true,
402
+ "audio_mode": "dmel",
403
+ "model_type": "inkling_audio",
404
+ "output_attentions": false
405
+ },
406
+ "tiny": {
407
+ "architectures": null,
408
+ "output_hidden_states": false,
409
+ "return_dict": true,
410
+ "dtype": null,
411
+ "chunk_size_feed_forward": 0,
412
+ "is_encoder_decoder": false,
413
+ "id2label": {
414
+ "0": "LABEL_0",
415
+ "1": "LABEL_1"
416
+ },
417
+ "label2id": {
418
+ "LABEL_0": 0,
419
+ "LABEL_1": 1
420
+ },
421
+ "problem_type": null,
422
+ "n_mel_bins": 80,
423
+ "mel_vocab_size": 16,
424
+ "text_hidden_size": 1536,
425
+ "rms_norm_eps": 1e-06,
426
+ "initializer_range": 0.02,
427
+ "_name_or_path": "",
428
+ "decoder_dmodel": 6144,
429
+ "bias": false,
430
+ "dmel_min_value": -7.0,
431
+ "dmel_max_value": 2.0,
432
+ "use_audio_norm": true,
433
+ "audio_mode": "dmel",
434
+ "model_type": "inkling_audio",
435
+ "output_attentions": false
436
+ }
437
+ },
438
+ "vision_config": {
439
+ "original": {
440
+ "architectures": null,
441
+ "output_hidden_states": false,
442
+ "return_dict": true,
443
+ "dtype": null,
444
+ "chunk_size_feed_forward": 0,
445
+ "is_encoder_decoder": false,
446
+ "id2label": {
447
+ "0": "LABEL_0",
448
+ "1": "LABEL_1"
449
+ },
450
+ "label2id": {
451
+ "LABEL_0": 0,
452
+ "LABEL_1": 1
453
+ },
454
+ "problem_type": null,
455
+ "text_hidden_size": 6144,
456
+ "patch_size": 40,
457
+ "temporal_patch_size": 2,
458
+ "num_channels": 3,
459
+ "hidden_size": 1024,
460
+ "n_layers": 4,
461
+ "num_attention_heads": 16,
462
+ "rms_norm_eps": 1e-06,
463
+ "initializer_range": 0.02,
464
+ "_name_or_path": "",
465
+ "vision_encoder_type": "hmlp",
466
+ "decoder_dmodel": 6144,
467
+ "n_channels": 3,
468
+ "use_vision_norm": true,
469
+ "model_type": "inkling_vision",
470
+ "output_attentions": false
471
+ },
472
+ "tiny": {
473
+ "architectures": null,
474
+ "output_hidden_states": false,
475
+ "return_dict": true,
476
+ "dtype": null,
477
+ "chunk_size_feed_forward": 0,
478
+ "is_encoder_decoder": false,
479
+ "id2label": {
480
+ "0": "LABEL_0",
481
+ "1": "LABEL_1"
482
+ },
483
+ "label2id": {
484
+ "LABEL_0": 0,
485
+ "LABEL_1": 1
486
+ },
487
+ "problem_type": null,
488
+ "text_hidden_size": 1536,
489
+ "patch_size": 40,
490
+ "temporal_patch_size": 2,
491
+ "num_channels": 3,
492
+ "hidden_size": 1024,
493
+ "n_layers": 4,
494
+ "num_attention_heads": 16,
495
+ "rms_norm_eps": 1e-06,
496
+ "initializer_range": 0.02,
497
+ "_name_or_path": "",
498
+ "vision_encoder_type": "hmlp",
499
+ "decoder_dmodel": 6144,
500
+ "n_channels": 3,
501
+ "use_vision_norm": true,
502
+ "model_type": "inkling_vision",
503
+ "output_attentions": false
504
+ }
505
+ },
506
+ "mtp_config": {
507
+ "original": {
508
+ "num_nextn_predict_layers": 8,
509
+ "chain_hidden_post_norm": false,
510
+ "local_layer_ids": [
511
+ 0,
512
+ 2,
513
+ 4,
514
+ 5,
515
+ 6,
516
+ 7
517
+ ]
518
+ },
519
+ "tiny": {
520
+ "num_nextn_predict_layers": 2,
521
+ "chain_hidden_post_norm": false,
522
+ "local_layer_ids": [
523
+ 0
524
+ ]
525
+ }
526
+ }
527
+ }
528
+ },
529
+ "backbone_validation": {
530
+ "name": "inkling",
531
+ "passed": true,
532
+ "reloaded_perplexity": 1.570810437713275,
533
+ "model_patches": false,
534
+ "backbone_parameters": 830199043,
535
+ "backbone_activated_parameters_estimate": 783013123,
536
+ "generation": "The capital of France is Paris. Paris is the largest city in France and serves as the country's political,"
537
+ },
538
+ "mtp_training": "synthetic initialized projections with trained final decoder block copied; not separately trained",
539
+ "training": {
540
+ "seed": 3225,
541
+ "initialization": "random",
542
+ "optimizer": "AdamW",
543
+ "learning_rate": 0.0004,
544
+ "weight_decay": 0.01,
545
+ "batch_size": 2,
546
+ "max_sequence_length": 160,
547
+ "evaluation": "same toy corpus as training"
548
+ }
549
+ }