joowon-jang commited on
Commit
b65a216
·
verified ·
1 Parent(s): ca08fec

Upload native MLX-VLM 8bit MTP variant

Browse files
.gitattributes CHANGED
@@ -33,3 +33,5 @@ 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
+ figures/Nex-N2-Benchmark-white.png filter=lfs diff=lfs merge=lfs -text
37
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,94 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: nex-agi/Nex-N2-mini
4
+ library_name: mlx
5
+ pipeline_tag: image-text-to-text
6
+ tags:
7
+ - mlx
8
+ - mlx-vlm
9
+ - safetensors
10
+ - apple-silicon
11
+ - vision-language
12
+ - image-text-to-text
13
+ - qwen3_5_moe
14
+ - qwen3.5
15
+ - nex-n2
16
+ - mtp
17
+ - native-mtp
18
+ - speculative-decoding
19
+ - omlx
20
+ - 8-bit-precision
21
+ - conversational
22
+ ---
23
+
24
+ # Nex-N2-mini-MLX-VLM-8bit-MTP
25
+
26
+ Native MLX-VLM 8-bit quantized version of `nex-agi/Nex-N2-mini`, with a grafted MTP head for oMLX Native MTP speculative decoding.
27
+
28
+ ## Summary
29
+
30
+ * Base model: `nex-agi/Nex-N2-mini`
31
+ * Format: native MLX / MLX-VLM
32
+ * Quantization: 8-bit MLX-VLM quantization
33
+ * Vision: supported
34
+ * MTP: included
35
+ * Target runtime: oMLX with Native MTP enabled
36
+ * Direct `mlx-vlm.generate`: not the supported runtime for this MTP variant
37
+
38
+ ## What changed in this version
39
+
40
+ This repository uses the native MLX-VLM trunk and vision weights from `Nex-N2-mini-MLX-VLM-8bit`, while the MTP head is grafted from the `jedisct1/Nex-N2-mini-mlx-OptiQ-8bit-MTP` MTP variant.
41
+
42
+ This is not the same as `jedisct1/Nex-N2-mini-mlx-OptiQ-8bit-MTP`:
43
+
44
+ * This repo keeps the native MLX-VLM layout.
45
+ * This repo includes the vision tower / VLM weights.
46
+ * The `jedisct1` OptiQ MTP repository is text-only.
47
+ * The MTP head is used only for speculative decoding acceleration.
48
+
49
+ ## Important compatibility note
50
+
51
+ This MTP variant includes `language_model.mtp.*` weights intended for oMLX Native MTP.
52
+
53
+ The validated runtime is:
54
+
55
+ * oMLX
56
+ * Native MTP enabled in model settings
57
+
58
+ Plain `mlx-vlm.generate` or generic MLX loading paths may fail strict loading or may not use the MTP head correctly, because the preserved `language_model.mtp.*` tensors are intended for the oMLX Native MTP runtime.
59
+
60
+ For direct `mlx-vlm.generate` usage, use the non-MTP variant instead:
61
+
62
+ `joowon-jang/Nex-N2-mini-MLX-VLM-8bit`
63
+
64
+ ## Quality and behavior
65
+
66
+ The MTP head is used for speculative decoding. Draft tokens are verified by the main Nex-N2-mini trunk before being accepted, so the MTP head is intended to affect speed rather than final output quality.
67
+
68
+ In practice, MTP speedups depend on draft acceptance rate. The grafted head tends to help more on normal prose and reasoning, and less on unusual token sequences.
69
+
70
+ ## Recommended runtime
71
+
72
+ Use oMLX and enable Native MTP.
73
+
74
+ Suggested initial settings:
75
+
76
+ * Native MTP: enabled
77
+ * Max Draft Tokens: 2
78
+ * Min Draft Tokens: 1
79
+ * Temperature: 0 for benchmarking
80
+ * Use the same prompt, context length, and max tokens when comparing against non-MTP variants
81
+
82
+ ## Notes
83
+
84
+ This is not an OptiQ oQ8 sidecar model. The model uses a native MLX-VLM layout with `vision_tower.*` weights included in the model files.
85
+
86
+ MTP head attribution:
87
+
88
+ * MTP head source: `jedisct1/Nex-N2-mini-mlx-OptiQ-8bit-MTP`
89
+ * Original base model: `nex-agi/Nex-N2-mini`
90
+ * Donor MTP lineage described by the source model: Qwen3.5-35B-A3B MTP head grafted onto Nex-N2-mini-compatible dimensions
91
+
92
+ ## License
93
+
94
+ Apache-2.0, following the base model and referenced MTP-head source license.
chat_template.jinja ADDED
@@ -0,0 +1,154 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- set image_count = namespace(value=0) %}
2
+ {%- set video_count = namespace(value=0) %}
3
+ {%- macro render_content(content, do_vision_count, is_system_content=false) %}
4
+ {%- if content is string %}
5
+ {{- content }}
6
+ {%- elif content is iterable and content is not mapping %}
7
+ {%- for item in content %}
8
+ {%- if 'image' in item or 'image_url' in item or item.type == 'image' %}
9
+ {%- if is_system_content %}
10
+ {{- raise_exception('System message cannot contain images.') }}
11
+ {%- endif %}
12
+ {%- if do_vision_count %}
13
+ {%- set image_count.value = image_count.value + 1 %}
14
+ {%- endif %}
15
+ {%- if add_vision_id %}
16
+ {{- 'Picture ' ~ image_count.value ~ ': ' }}
17
+ {%- endif %}
18
+ {{- '<|vision_start|><|image_pad|><|vision_end|>' }}
19
+ {%- elif 'video' in item or item.type == 'video' %}
20
+ {%- if is_system_content %}
21
+ {{- raise_exception('System message cannot contain videos.') }}
22
+ {%- endif %}
23
+ {%- if do_vision_count %}
24
+ {%- set video_count.value = video_count.value + 1 %}
25
+ {%- endif %}
26
+ {%- if add_vision_id %}
27
+ {{- 'Video ' ~ video_count.value ~ ': ' }}
28
+ {%- endif %}
29
+ {{- '<|vision_start|><|video_pad|><|vision_end|>' }}
30
+ {%- elif 'text' in item %}
31
+ {{- item.text }}
32
+ {%- else %}
33
+ {{- raise_exception('Unexpected item type in content.') }}
34
+ {%- endif %}
35
+ {%- endfor %}
36
+ {%- elif content is none or content is undefined %}
37
+ {{- '' }}
38
+ {%- else %}
39
+ {{- raise_exception('Unexpected content type.') }}
40
+ {%- endif %}
41
+ {%- endmacro %}
42
+ {%- if not messages %}
43
+ {{- raise_exception('No messages provided.') }}
44
+ {%- endif %}
45
+ {%- if tools and tools is iterable and tools is not mapping %}
46
+ {{- '<|im_start|>system\n' }}
47
+ {{- "# Tools\n\nYou have access to the following functions:\n\n<tools>" }}
48
+ {%- for tool in tools %}
49
+ {{- "\n" }}
50
+ {{- tool | tojson }}
51
+ {%- endfor %}
52
+ {{- "\n</tools>" }}
53
+ {{- '\n\nIf you choose to call a function ONLY reply in the following format with NO suffix:\n\n<tool_call>\n<function=example_function_name>\n<parameter=example_parameter_1>\nvalue_1\n</parameter>\n<parameter=example_parameter_2>\nThis is the value for the second parameter\nthat can span\nmultiple lines\n</parameter>\n</function>\n</tool_call>\n\n<IMPORTANT>\nReminder:\n- Function calls MUST follow the specified format: an inner <function=...></function> block must be nested within <tool_call></tool_call> XML tags\n- Required parameters MUST be specified\n- You may provide optional reasoning for your function call in natural language BEFORE the function call, but NOT after\n- If there is no function call available, answer the question like normal with your current knowledge and do not tell the user about function calls\n</IMPORTANT>' }}
54
+ {%- if messages[0].role == 'system' %}
55
+ {%- set content = render_content(messages[0].content, false, true)|trim %}
56
+ {%- if content %}
57
+ {{- '\n\n' + content }}
58
+ {%- endif %}
59
+ {%- endif %}
60
+ {{- '<|im_end|>\n' }}
61
+ {%- else %}
62
+ {%- if messages[0].role == 'system' %}
63
+ {%- set content = render_content(messages[0].content, false, true)|trim %}
64
+ {{- '<|im_start|>system\n' + content + '<|im_end|>\n' }}
65
+ {%- endif %}
66
+ {%- endif %}
67
+ {%- set ns = namespace(multi_step_tool=true, last_query_index=messages|length - 1) %}
68
+ {%- for message in messages[::-1] %}
69
+ {%- set index = (messages|length - 1) - loop.index0 %}
70
+ {%- if ns.multi_step_tool and message.role == "user" %}
71
+ {%- set content = render_content(message.content, false)|trim %}
72
+ {%- if not(content.startswith('<tool_response>') and content.endswith('</tool_response>')) %}
73
+ {%- set ns.multi_step_tool = false %}
74
+ {%- set ns.last_query_index = index %}
75
+ {%- endif %}
76
+ {%- endif %}
77
+ {%- endfor %}
78
+ {%- if ns.multi_step_tool %}
79
+ {{- raise_exception('No user query found in messages.') }}
80
+ {%- endif %}
81
+ {%- for message in messages %}
82
+ {%- set content = render_content(message.content, true)|trim %}
83
+ {%- if message.role == "system" %}
84
+ {%- if not loop.first %}
85
+ {{- '<|im_start|>' + message.role + '\n' + content + '<|im_end|>' + '\n' }}
86
+ {%- endif %}
87
+ {%- elif message.role == "user" %}
88
+ {{- '<|im_start|>' + message.role + '\n' + content + '<|im_end|>' + '\n' }}
89
+ {%- elif message.role == "assistant" %}
90
+ {%- set reasoning_content = '' %}
91
+ {%- if message.reasoning_content is string %}
92
+ {%- set reasoning_content = message.reasoning_content %}
93
+ {%- else %}
94
+ {%- if '</think>' in content %}
95
+ {%- set reasoning_content = content.split('</think>')[0].rstrip('\n').split('<think>')[-1].lstrip('\n') %}
96
+ {%- set content = content.split('</think>')[-1].lstrip('\n') %}
97
+ {%- endif %}
98
+ {%- endif %}
99
+ {%- set reasoning_content = reasoning_content|trim %}
100
+ {%- if reasoning_content %}
101
+ {{- '<|im_start|>' + message.role + '\n<think>\n' + reasoning_content + '\n</think>\n\n' + content }}
102
+ {%- else %}
103
+ {{- '<|im_start|>' + message.role + '\n' + content }}
104
+ {%- endif %}
105
+ {%- if message.tool_calls and message.tool_calls is iterable and message.tool_calls is not mapping %}
106
+ {%- for tool_call in message.tool_calls %}
107
+ {%- if tool_call.function is defined %}
108
+ {%- set tool_call = tool_call.function %}
109
+ {%- endif %}
110
+ {%- if loop.first %}
111
+ {%- if content|trim %}
112
+ {{- '\n\n<tool_call>\n<function=' + tool_call.name + '>\n' }}
113
+ {%- else %}
114
+ {{- '<tool_call>\n<function=' + tool_call.name + '>\n' }}
115
+ {%- endif %}
116
+ {%- else %}
117
+ {{- '\n<tool_call>\n<function=' + tool_call.name + '>\n' }}
118
+ {%- endif %}
119
+ {%- if tool_call.arguments is defined %}
120
+ {%- for args_name, args_value in tool_call.arguments|items %}
121
+ {{- '<parameter=' + args_name + '>\n' }}
122
+ {%- set args_value = args_value | tojson | safe if args_value is mapping or (args_value is sequence and args_value is not string) else args_value | string %}
123
+ {{- args_value }}
124
+ {{- '\n</parameter>\n' }}
125
+ {%- endfor %}
126
+ {%- endif %}
127
+ {{- '</function>\n</tool_call>' }}
128
+ {%- endfor %}
129
+ {%- endif %}
130
+ {{- '<|im_end|>\n' }}
131
+ {%- elif message.role == "tool" %}
132
+ {%- if loop.previtem and loop.previtem.role != "tool" %}
133
+ {{- '<|im_start|>user' }}
134
+ {%- endif %}
135
+ {{- '\n<tool_response>\n' }}
136
+ {{- content }}
137
+ {{- '\n</tool_response>' }}
138
+ {%- if not loop.last and loop.nextitem.role != "tool" %}
139
+ {{- '<|im_end|>\n' }}
140
+ {%- elif loop.last %}
141
+ {{- '<|im_end|>\n' }}
142
+ {%- endif %}
143
+ {%- else %}
144
+ {{- raise_exception('Unexpected message role.') }}
145
+ {%- endif %}
146
+ {%- endfor %}
147
+ {%- if add_generation_prompt %}
148
+ {{- '<|im_start|>assistant\n' }}
149
+ {%- if enable_thinking is defined and enable_thinking is false %}
150
+ {{- '<think>\n\n</think>\n\n' }}
151
+ {%- else %}
152
+ {{- '<think>' }}
153
+ {%- endif %}
154
+ {%- endif %}
config.json ADDED
@@ -0,0 +1,901 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Qwen3_5MoeForConditionalGeneration"
4
+ ],
5
+ "dtype": "bfloat16",
6
+ "hidden_size": 2048,
7
+ "image_token_id": 248056,
8
+ "model_type": "qwen3_5_moe",
9
+ "quantization": {
10
+ "group_size": 64,
11
+ "bits": 8,
12
+ "mode": "affine",
13
+ "language_model.model.layers.0.mlp.gate": {
14
+ "group_size": 64,
15
+ "bits": 8
16
+ },
17
+ "language_model.model.layers.0.mlp.shared_expert_gate": {
18
+ "group_size": 64,
19
+ "bits": 8
20
+ },
21
+ "language_model.model.layers.1.mlp.gate": {
22
+ "group_size": 64,
23
+ "bits": 8
24
+ },
25
+ "language_model.model.layers.1.mlp.shared_expert_gate": {
26
+ "group_size": 64,
27
+ "bits": 8
28
+ },
29
+ "language_model.model.layers.2.mlp.gate": {
30
+ "group_size": 64,
31
+ "bits": 8
32
+ },
33
+ "language_model.model.layers.2.mlp.shared_expert_gate": {
34
+ "group_size": 64,
35
+ "bits": 8
36
+ },
37
+ "language_model.model.layers.3.mlp.gate": {
38
+ "group_size": 64,
39
+ "bits": 8
40
+ },
41
+ "language_model.model.layers.3.mlp.shared_expert_gate": {
42
+ "group_size": 64,
43
+ "bits": 8
44
+ },
45
+ "language_model.model.layers.4.mlp.gate": {
46
+ "group_size": 64,
47
+ "bits": 8
48
+ },
49
+ "language_model.model.layers.4.mlp.shared_expert_gate": {
50
+ "group_size": 64,
51
+ "bits": 8
52
+ },
53
+ "language_model.model.layers.5.mlp.gate": {
54
+ "group_size": 64,
55
+ "bits": 8
56
+ },
57
+ "language_model.model.layers.5.mlp.shared_expert_gate": {
58
+ "group_size": 64,
59
+ "bits": 8
60
+ },
61
+ "language_model.model.layers.6.mlp.gate": {
62
+ "group_size": 64,
63
+ "bits": 8
64
+ },
65
+ "language_model.model.layers.6.mlp.shared_expert_gate": {
66
+ "group_size": 64,
67
+ "bits": 8
68
+ },
69
+ "language_model.model.layers.7.mlp.gate": {
70
+ "group_size": 64,
71
+ "bits": 8
72
+ },
73
+ "language_model.model.layers.7.mlp.shared_expert_gate": {
74
+ "group_size": 64,
75
+ "bits": 8
76
+ },
77
+ "language_model.model.layers.8.mlp.gate": {
78
+ "group_size": 64,
79
+ "bits": 8
80
+ },
81
+ "language_model.model.layers.8.mlp.shared_expert_gate": {
82
+ "group_size": 64,
83
+ "bits": 8
84
+ },
85
+ "language_model.model.layers.9.mlp.gate": {
86
+ "group_size": 64,
87
+ "bits": 8
88
+ },
89
+ "language_model.model.layers.9.mlp.shared_expert_gate": {
90
+ "group_size": 64,
91
+ "bits": 8
92
+ },
93
+ "language_model.model.layers.10.mlp.gate": {
94
+ "group_size": 64,
95
+ "bits": 8
96
+ },
97
+ "language_model.model.layers.10.mlp.shared_expert_gate": {
98
+ "group_size": 64,
99
+ "bits": 8
100
+ },
101
+ "language_model.model.layers.11.mlp.gate": {
102
+ "group_size": 64,
103
+ "bits": 8
104
+ },
105
+ "language_model.model.layers.11.mlp.shared_expert_gate": {
106
+ "group_size": 64,
107
+ "bits": 8
108
+ },
109
+ "language_model.model.layers.12.mlp.gate": {
110
+ "group_size": 64,
111
+ "bits": 8
112
+ },
113
+ "language_model.model.layers.12.mlp.shared_expert_gate": {
114
+ "group_size": 64,
115
+ "bits": 8
116
+ },
117
+ "language_model.model.layers.13.mlp.gate": {
118
+ "group_size": 64,
119
+ "bits": 8
120
+ },
121
+ "language_model.model.layers.13.mlp.shared_expert_gate": {
122
+ "group_size": 64,
123
+ "bits": 8
124
+ },
125
+ "language_model.model.layers.14.mlp.gate": {
126
+ "group_size": 64,
127
+ "bits": 8
128
+ },
129
+ "language_model.model.layers.14.mlp.shared_expert_gate": {
130
+ "group_size": 64,
131
+ "bits": 8
132
+ },
133
+ "language_model.model.layers.15.mlp.gate": {
134
+ "group_size": 64,
135
+ "bits": 8
136
+ },
137
+ "language_model.model.layers.15.mlp.shared_expert_gate": {
138
+ "group_size": 64,
139
+ "bits": 8
140
+ },
141
+ "language_model.model.layers.16.mlp.gate": {
142
+ "group_size": 64,
143
+ "bits": 8
144
+ },
145
+ "language_model.model.layers.16.mlp.shared_expert_gate": {
146
+ "group_size": 64,
147
+ "bits": 8
148
+ },
149
+ "language_model.model.layers.17.mlp.gate": {
150
+ "group_size": 64,
151
+ "bits": 8
152
+ },
153
+ "language_model.model.layers.17.mlp.shared_expert_gate": {
154
+ "group_size": 64,
155
+ "bits": 8
156
+ },
157
+ "language_model.model.layers.18.mlp.gate": {
158
+ "group_size": 64,
159
+ "bits": 8
160
+ },
161
+ "language_model.model.layers.18.mlp.shared_expert_gate": {
162
+ "group_size": 64,
163
+ "bits": 8
164
+ },
165
+ "language_model.model.layers.19.mlp.gate": {
166
+ "group_size": 64,
167
+ "bits": 8
168
+ },
169
+ "language_model.model.layers.19.mlp.shared_expert_gate": {
170
+ "group_size": 64,
171
+ "bits": 8
172
+ },
173
+ "language_model.model.layers.20.mlp.gate": {
174
+ "group_size": 64,
175
+ "bits": 8
176
+ },
177
+ "language_model.model.layers.20.mlp.shared_expert_gate": {
178
+ "group_size": 64,
179
+ "bits": 8
180
+ },
181
+ "language_model.model.layers.21.mlp.gate": {
182
+ "group_size": 64,
183
+ "bits": 8
184
+ },
185
+ "language_model.model.layers.21.mlp.shared_expert_gate": {
186
+ "group_size": 64,
187
+ "bits": 8
188
+ },
189
+ "language_model.model.layers.22.mlp.gate": {
190
+ "group_size": 64,
191
+ "bits": 8
192
+ },
193
+ "language_model.model.layers.22.mlp.shared_expert_gate": {
194
+ "group_size": 64,
195
+ "bits": 8
196
+ },
197
+ "language_model.model.layers.23.mlp.gate": {
198
+ "group_size": 64,
199
+ "bits": 8
200
+ },
201
+ "language_model.model.layers.23.mlp.shared_expert_gate": {
202
+ "group_size": 64,
203
+ "bits": 8
204
+ },
205
+ "language_model.model.layers.24.mlp.gate": {
206
+ "group_size": 64,
207
+ "bits": 8
208
+ },
209
+ "language_model.model.layers.24.mlp.shared_expert_gate": {
210
+ "group_size": 64,
211
+ "bits": 8
212
+ },
213
+ "language_model.model.layers.25.mlp.gate": {
214
+ "group_size": 64,
215
+ "bits": 8
216
+ },
217
+ "language_model.model.layers.25.mlp.shared_expert_gate": {
218
+ "group_size": 64,
219
+ "bits": 8
220
+ },
221
+ "language_model.model.layers.26.mlp.gate": {
222
+ "group_size": 64,
223
+ "bits": 8
224
+ },
225
+ "language_model.model.layers.26.mlp.shared_expert_gate": {
226
+ "group_size": 64,
227
+ "bits": 8
228
+ },
229
+ "language_model.model.layers.27.mlp.gate": {
230
+ "group_size": 64,
231
+ "bits": 8
232
+ },
233
+ "language_model.model.layers.27.mlp.shared_expert_gate": {
234
+ "group_size": 64,
235
+ "bits": 8
236
+ },
237
+ "language_model.model.layers.28.mlp.gate": {
238
+ "group_size": 64,
239
+ "bits": 8
240
+ },
241
+ "language_model.model.layers.28.mlp.shared_expert_gate": {
242
+ "group_size": 64,
243
+ "bits": 8
244
+ },
245
+ "language_model.model.layers.29.mlp.gate": {
246
+ "group_size": 64,
247
+ "bits": 8
248
+ },
249
+ "language_model.model.layers.29.mlp.shared_expert_gate": {
250
+ "group_size": 64,
251
+ "bits": 8
252
+ },
253
+ "language_model.model.layers.30.mlp.gate": {
254
+ "group_size": 64,
255
+ "bits": 8
256
+ },
257
+ "language_model.model.layers.30.mlp.shared_expert_gate": {
258
+ "group_size": 64,
259
+ "bits": 8
260
+ },
261
+ "language_model.model.layers.31.mlp.gate": {
262
+ "group_size": 64,
263
+ "bits": 8
264
+ },
265
+ "language_model.model.layers.31.mlp.shared_expert_gate": {
266
+ "group_size": 64,
267
+ "bits": 8
268
+ },
269
+ "language_model.model.layers.32.mlp.gate": {
270
+ "group_size": 64,
271
+ "bits": 8
272
+ },
273
+ "language_model.model.layers.32.mlp.shared_expert_gate": {
274
+ "group_size": 64,
275
+ "bits": 8
276
+ },
277
+ "language_model.model.layers.33.mlp.gate": {
278
+ "group_size": 64,
279
+ "bits": 8
280
+ },
281
+ "language_model.model.layers.33.mlp.shared_expert_gate": {
282
+ "group_size": 64,
283
+ "bits": 8
284
+ },
285
+ "language_model.model.layers.34.mlp.gate": {
286
+ "group_size": 64,
287
+ "bits": 8
288
+ },
289
+ "language_model.model.layers.34.mlp.shared_expert_gate": {
290
+ "group_size": 64,
291
+ "bits": 8
292
+ },
293
+ "language_model.model.layers.35.mlp.gate": {
294
+ "group_size": 64,
295
+ "bits": 8
296
+ },
297
+ "language_model.model.layers.35.mlp.shared_expert_gate": {
298
+ "group_size": 64,
299
+ "bits": 8
300
+ },
301
+ "language_model.model.layers.36.mlp.gate": {
302
+ "group_size": 64,
303
+ "bits": 8
304
+ },
305
+ "language_model.model.layers.36.mlp.shared_expert_gate": {
306
+ "group_size": 64,
307
+ "bits": 8
308
+ },
309
+ "language_model.model.layers.37.mlp.gate": {
310
+ "group_size": 64,
311
+ "bits": 8
312
+ },
313
+ "language_model.model.layers.37.mlp.shared_expert_gate": {
314
+ "group_size": 64,
315
+ "bits": 8
316
+ },
317
+ "language_model.model.layers.38.mlp.gate": {
318
+ "group_size": 64,
319
+ "bits": 8
320
+ },
321
+ "language_model.model.layers.38.mlp.shared_expert_gate": {
322
+ "group_size": 64,
323
+ "bits": 8
324
+ },
325
+ "language_model.model.layers.39.mlp.gate": {
326
+ "group_size": 64,
327
+ "bits": 8
328
+ },
329
+ "language_model.model.layers.39.mlp.shared_expert_gate": {
330
+ "group_size": 64,
331
+ "bits": 8
332
+ },
333
+ "language_model.mtp.layers.0.mlp.shared_expert.down_proj": {
334
+ "bits": 8,
335
+ "group_size": 128,
336
+ "mode": "affine"
337
+ },
338
+ "language_model.mtp.layers.0.mlp.shared_expert.gate_proj": {
339
+ "bits": 8,
340
+ "group_size": 128,
341
+ "mode": "affine"
342
+ },
343
+ "language_model.mtp.layers.0.mlp.shared_expert.up_proj": {
344
+ "bits": 8,
345
+ "group_size": 128,
346
+ "mode": "affine"
347
+ },
348
+ "language_model.mtp.layers.0.self_attn.k_proj": {
349
+ "bits": 8,
350
+ "group_size": 128,
351
+ "mode": "affine"
352
+ },
353
+ "language_model.mtp.layers.0.self_attn.q_proj": {
354
+ "bits": 8,
355
+ "group_size": 128,
356
+ "mode": "affine"
357
+ },
358
+ "language_model.mtp.layers.0.self_attn.v_proj": {
359
+ "bits": 8,
360
+ "group_size": 128,
361
+ "mode": "affine"
362
+ }
363
+ },
364
+ "quantization_config": {
365
+ "group_size": 64,
366
+ "bits": 8,
367
+ "mode": "affine",
368
+ "language_model.model.layers.0.mlp.gate": {
369
+ "group_size": 64,
370
+ "bits": 8
371
+ },
372
+ "language_model.model.layers.0.mlp.shared_expert_gate": {
373
+ "group_size": 64,
374
+ "bits": 8
375
+ },
376
+ "language_model.model.layers.1.mlp.gate": {
377
+ "group_size": 64,
378
+ "bits": 8
379
+ },
380
+ "language_model.model.layers.1.mlp.shared_expert_gate": {
381
+ "group_size": 64,
382
+ "bits": 8
383
+ },
384
+ "language_model.model.layers.2.mlp.gate": {
385
+ "group_size": 64,
386
+ "bits": 8
387
+ },
388
+ "language_model.model.layers.2.mlp.shared_expert_gate": {
389
+ "group_size": 64,
390
+ "bits": 8
391
+ },
392
+ "language_model.model.layers.3.mlp.gate": {
393
+ "group_size": 64,
394
+ "bits": 8
395
+ },
396
+ "language_model.model.layers.3.mlp.shared_expert_gate": {
397
+ "group_size": 64,
398
+ "bits": 8
399
+ },
400
+ "language_model.model.layers.4.mlp.gate": {
401
+ "group_size": 64,
402
+ "bits": 8
403
+ },
404
+ "language_model.model.layers.4.mlp.shared_expert_gate": {
405
+ "group_size": 64,
406
+ "bits": 8
407
+ },
408
+ "language_model.model.layers.5.mlp.gate": {
409
+ "group_size": 64,
410
+ "bits": 8
411
+ },
412
+ "language_model.model.layers.5.mlp.shared_expert_gate": {
413
+ "group_size": 64,
414
+ "bits": 8
415
+ },
416
+ "language_model.model.layers.6.mlp.gate": {
417
+ "group_size": 64,
418
+ "bits": 8
419
+ },
420
+ "language_model.model.layers.6.mlp.shared_expert_gate": {
421
+ "group_size": 64,
422
+ "bits": 8
423
+ },
424
+ "language_model.model.layers.7.mlp.gate": {
425
+ "group_size": 64,
426
+ "bits": 8
427
+ },
428
+ "language_model.model.layers.7.mlp.shared_expert_gate": {
429
+ "group_size": 64,
430
+ "bits": 8
431
+ },
432
+ "language_model.model.layers.8.mlp.gate": {
433
+ "group_size": 64,
434
+ "bits": 8
435
+ },
436
+ "language_model.model.layers.8.mlp.shared_expert_gate": {
437
+ "group_size": 64,
438
+ "bits": 8
439
+ },
440
+ "language_model.model.layers.9.mlp.gate": {
441
+ "group_size": 64,
442
+ "bits": 8
443
+ },
444
+ "language_model.model.layers.9.mlp.shared_expert_gate": {
445
+ "group_size": 64,
446
+ "bits": 8
447
+ },
448
+ "language_model.model.layers.10.mlp.gate": {
449
+ "group_size": 64,
450
+ "bits": 8
451
+ },
452
+ "language_model.model.layers.10.mlp.shared_expert_gate": {
453
+ "group_size": 64,
454
+ "bits": 8
455
+ },
456
+ "language_model.model.layers.11.mlp.gate": {
457
+ "group_size": 64,
458
+ "bits": 8
459
+ },
460
+ "language_model.model.layers.11.mlp.shared_expert_gate": {
461
+ "group_size": 64,
462
+ "bits": 8
463
+ },
464
+ "language_model.model.layers.12.mlp.gate": {
465
+ "group_size": 64,
466
+ "bits": 8
467
+ },
468
+ "language_model.model.layers.12.mlp.shared_expert_gate": {
469
+ "group_size": 64,
470
+ "bits": 8
471
+ },
472
+ "language_model.model.layers.13.mlp.gate": {
473
+ "group_size": 64,
474
+ "bits": 8
475
+ },
476
+ "language_model.model.layers.13.mlp.shared_expert_gate": {
477
+ "group_size": 64,
478
+ "bits": 8
479
+ },
480
+ "language_model.model.layers.14.mlp.gate": {
481
+ "group_size": 64,
482
+ "bits": 8
483
+ },
484
+ "language_model.model.layers.14.mlp.shared_expert_gate": {
485
+ "group_size": 64,
486
+ "bits": 8
487
+ },
488
+ "language_model.model.layers.15.mlp.gate": {
489
+ "group_size": 64,
490
+ "bits": 8
491
+ },
492
+ "language_model.model.layers.15.mlp.shared_expert_gate": {
493
+ "group_size": 64,
494
+ "bits": 8
495
+ },
496
+ "language_model.model.layers.16.mlp.gate": {
497
+ "group_size": 64,
498
+ "bits": 8
499
+ },
500
+ "language_model.model.layers.16.mlp.shared_expert_gate": {
501
+ "group_size": 64,
502
+ "bits": 8
503
+ },
504
+ "language_model.model.layers.17.mlp.gate": {
505
+ "group_size": 64,
506
+ "bits": 8
507
+ },
508
+ "language_model.model.layers.17.mlp.shared_expert_gate": {
509
+ "group_size": 64,
510
+ "bits": 8
511
+ },
512
+ "language_model.model.layers.18.mlp.gate": {
513
+ "group_size": 64,
514
+ "bits": 8
515
+ },
516
+ "language_model.model.layers.18.mlp.shared_expert_gate": {
517
+ "group_size": 64,
518
+ "bits": 8
519
+ },
520
+ "language_model.model.layers.19.mlp.gate": {
521
+ "group_size": 64,
522
+ "bits": 8
523
+ },
524
+ "language_model.model.layers.19.mlp.shared_expert_gate": {
525
+ "group_size": 64,
526
+ "bits": 8
527
+ },
528
+ "language_model.model.layers.20.mlp.gate": {
529
+ "group_size": 64,
530
+ "bits": 8
531
+ },
532
+ "language_model.model.layers.20.mlp.shared_expert_gate": {
533
+ "group_size": 64,
534
+ "bits": 8
535
+ },
536
+ "language_model.model.layers.21.mlp.gate": {
537
+ "group_size": 64,
538
+ "bits": 8
539
+ },
540
+ "language_model.model.layers.21.mlp.shared_expert_gate": {
541
+ "group_size": 64,
542
+ "bits": 8
543
+ },
544
+ "language_model.model.layers.22.mlp.gate": {
545
+ "group_size": 64,
546
+ "bits": 8
547
+ },
548
+ "language_model.model.layers.22.mlp.shared_expert_gate": {
549
+ "group_size": 64,
550
+ "bits": 8
551
+ },
552
+ "language_model.model.layers.23.mlp.gate": {
553
+ "group_size": 64,
554
+ "bits": 8
555
+ },
556
+ "language_model.model.layers.23.mlp.shared_expert_gate": {
557
+ "group_size": 64,
558
+ "bits": 8
559
+ },
560
+ "language_model.model.layers.24.mlp.gate": {
561
+ "group_size": 64,
562
+ "bits": 8
563
+ },
564
+ "language_model.model.layers.24.mlp.shared_expert_gate": {
565
+ "group_size": 64,
566
+ "bits": 8
567
+ },
568
+ "language_model.model.layers.25.mlp.gate": {
569
+ "group_size": 64,
570
+ "bits": 8
571
+ },
572
+ "language_model.model.layers.25.mlp.shared_expert_gate": {
573
+ "group_size": 64,
574
+ "bits": 8
575
+ },
576
+ "language_model.model.layers.26.mlp.gate": {
577
+ "group_size": 64,
578
+ "bits": 8
579
+ },
580
+ "language_model.model.layers.26.mlp.shared_expert_gate": {
581
+ "group_size": 64,
582
+ "bits": 8
583
+ },
584
+ "language_model.model.layers.27.mlp.gate": {
585
+ "group_size": 64,
586
+ "bits": 8
587
+ },
588
+ "language_model.model.layers.27.mlp.shared_expert_gate": {
589
+ "group_size": 64,
590
+ "bits": 8
591
+ },
592
+ "language_model.model.layers.28.mlp.gate": {
593
+ "group_size": 64,
594
+ "bits": 8
595
+ },
596
+ "language_model.model.layers.28.mlp.shared_expert_gate": {
597
+ "group_size": 64,
598
+ "bits": 8
599
+ },
600
+ "language_model.model.layers.29.mlp.gate": {
601
+ "group_size": 64,
602
+ "bits": 8
603
+ },
604
+ "language_model.model.layers.29.mlp.shared_expert_gate": {
605
+ "group_size": 64,
606
+ "bits": 8
607
+ },
608
+ "language_model.model.layers.30.mlp.gate": {
609
+ "group_size": 64,
610
+ "bits": 8
611
+ },
612
+ "language_model.model.layers.30.mlp.shared_expert_gate": {
613
+ "group_size": 64,
614
+ "bits": 8
615
+ },
616
+ "language_model.model.layers.31.mlp.gate": {
617
+ "group_size": 64,
618
+ "bits": 8
619
+ },
620
+ "language_model.model.layers.31.mlp.shared_expert_gate": {
621
+ "group_size": 64,
622
+ "bits": 8
623
+ },
624
+ "language_model.model.layers.32.mlp.gate": {
625
+ "group_size": 64,
626
+ "bits": 8
627
+ },
628
+ "language_model.model.layers.32.mlp.shared_expert_gate": {
629
+ "group_size": 64,
630
+ "bits": 8
631
+ },
632
+ "language_model.model.layers.33.mlp.gate": {
633
+ "group_size": 64,
634
+ "bits": 8
635
+ },
636
+ "language_model.model.layers.33.mlp.shared_expert_gate": {
637
+ "group_size": 64,
638
+ "bits": 8
639
+ },
640
+ "language_model.model.layers.34.mlp.gate": {
641
+ "group_size": 64,
642
+ "bits": 8
643
+ },
644
+ "language_model.model.layers.34.mlp.shared_expert_gate": {
645
+ "group_size": 64,
646
+ "bits": 8
647
+ },
648
+ "language_model.model.layers.35.mlp.gate": {
649
+ "group_size": 64,
650
+ "bits": 8
651
+ },
652
+ "language_model.model.layers.35.mlp.shared_expert_gate": {
653
+ "group_size": 64,
654
+ "bits": 8
655
+ },
656
+ "language_model.model.layers.36.mlp.gate": {
657
+ "group_size": 64,
658
+ "bits": 8
659
+ },
660
+ "language_model.model.layers.36.mlp.shared_expert_gate": {
661
+ "group_size": 64,
662
+ "bits": 8
663
+ },
664
+ "language_model.model.layers.37.mlp.gate": {
665
+ "group_size": 64,
666
+ "bits": 8
667
+ },
668
+ "language_model.model.layers.37.mlp.shared_expert_gate": {
669
+ "group_size": 64,
670
+ "bits": 8
671
+ },
672
+ "language_model.model.layers.38.mlp.gate": {
673
+ "group_size": 64,
674
+ "bits": 8
675
+ },
676
+ "language_model.model.layers.38.mlp.shared_expert_gate": {
677
+ "group_size": 64,
678
+ "bits": 8
679
+ },
680
+ "language_model.model.layers.39.mlp.gate": {
681
+ "group_size": 64,
682
+ "bits": 8
683
+ },
684
+ "language_model.model.layers.39.mlp.shared_expert_gate": {
685
+ "group_size": 64,
686
+ "bits": 8
687
+ },
688
+ "language_model.mtp.layers.0.mlp.shared_expert.down_proj": {
689
+ "bits": 8,
690
+ "group_size": 128,
691
+ "mode": "affine"
692
+ },
693
+ "language_model.mtp.layers.0.mlp.shared_expert.gate_proj": {
694
+ "bits": 8,
695
+ "group_size": 128,
696
+ "mode": "affine"
697
+ },
698
+ "language_model.mtp.layers.0.mlp.shared_expert.up_proj": {
699
+ "bits": 8,
700
+ "group_size": 128,
701
+ "mode": "affine"
702
+ },
703
+ "language_model.mtp.layers.0.self_attn.k_proj": {
704
+ "bits": 8,
705
+ "group_size": 128,
706
+ "mode": "affine"
707
+ },
708
+ "language_model.mtp.layers.0.self_attn.q_proj": {
709
+ "bits": 8,
710
+ "group_size": 128,
711
+ "mode": "affine"
712
+ },
713
+ "language_model.mtp.layers.0.self_attn.v_proj": {
714
+ "bits": 8,
715
+ "group_size": 128,
716
+ "mode": "affine"
717
+ }
718
+ },
719
+ "text_config": {
720
+ "attention_bias": false,
721
+ "attention_dropout": 0.0,
722
+ "attn_output_gate": true,
723
+ "bos_token_id": null,
724
+ "dtype": "bfloat16",
725
+ "eos_token_id": 248044,
726
+ "full_attention_interval": 4,
727
+ "head_dim": 256,
728
+ "hidden_act": "silu",
729
+ "hidden_size": 2048,
730
+ "initializer_range": 0.02,
731
+ "layer_types": [
732
+ "linear_attention",
733
+ "linear_attention",
734
+ "linear_attention",
735
+ "full_attention",
736
+ "linear_attention",
737
+ "linear_attention",
738
+ "linear_attention",
739
+ "full_attention",
740
+ "linear_attention",
741
+ "linear_attention",
742
+ "linear_attention",
743
+ "full_attention",
744
+ "linear_attention",
745
+ "linear_attention",
746
+ "linear_attention",
747
+ "full_attention",
748
+ "linear_attention",
749
+ "linear_attention",
750
+ "linear_attention",
751
+ "full_attention",
752
+ "linear_attention",
753
+ "linear_attention",
754
+ "linear_attention",
755
+ "full_attention",
756
+ "linear_attention",
757
+ "linear_attention",
758
+ "linear_attention",
759
+ "full_attention",
760
+ "linear_attention",
761
+ "linear_attention",
762
+ "linear_attention",
763
+ "full_attention",
764
+ "linear_attention",
765
+ "linear_attention",
766
+ "linear_attention",
767
+ "full_attention",
768
+ "linear_attention",
769
+ "linear_attention",
770
+ "linear_attention",
771
+ "full_attention"
772
+ ],
773
+ "linear_conv_kernel_dim": 4,
774
+ "linear_key_head_dim": 128,
775
+ "linear_num_key_heads": 16,
776
+ "linear_num_value_heads": 32,
777
+ "linear_value_head_dim": 128,
778
+ "mamba_ssm_dtype": "float32",
779
+ "max_position_embeddings": 262144,
780
+ "mlp_only_layers": [],
781
+ "model_type": "qwen3_5_moe_text",
782
+ "moe_intermediate_size": 512,
783
+ "mtp_num_hidden_layers": 1,
784
+ "mtp_use_dedicated_embeddings": false,
785
+ "num_attention_heads": 16,
786
+ "num_experts": 256,
787
+ "num_experts_per_tok": 8,
788
+ "num_hidden_layers": 40,
789
+ "num_key_value_heads": 2,
790
+ "output_router_logits": false,
791
+ "pad_token_id": null,
792
+ "partial_rotary_factor": 0.25,
793
+ "rms_norm_eps": 1e-06,
794
+ "rope_parameters": {
795
+ "mrope_interleaved": true,
796
+ "mrope_section": [
797
+ 11,
798
+ 11,
799
+ 10
800
+ ],
801
+ "partial_rotary_factor": 0.25,
802
+ "rope_theta": 10000000,
803
+ "rope_type": "default"
804
+ },
805
+ "router_aux_loss_coef": 0.001,
806
+ "shared_expert_intermediate_size": 512,
807
+ "tie_word_embeddings": false,
808
+ "use_cache": true,
809
+ "vocab_size": 248320,
810
+ "quantization": {
811
+ "bits": 8,
812
+ "group_size": 64,
813
+ "language_model.mtp.layers.0.mlp.shared_expert.down_proj": {
814
+ "bits": 8,
815
+ "group_size": 128,
816
+ "mode": "affine"
817
+ },
818
+ "language_model.mtp.layers.0.mlp.shared_expert.gate_proj": {
819
+ "bits": 8,
820
+ "group_size": 128,
821
+ "mode": "affine"
822
+ },
823
+ "language_model.mtp.layers.0.mlp.shared_expert.up_proj": {
824
+ "bits": 8,
825
+ "group_size": 128,
826
+ "mode": "affine"
827
+ },
828
+ "language_model.mtp.layers.0.self_attn.k_proj": {
829
+ "bits": 8,
830
+ "group_size": 128,
831
+ "mode": "affine"
832
+ },
833
+ "language_model.mtp.layers.0.self_attn.q_proj": {
834
+ "bits": 8,
835
+ "group_size": 128,
836
+ "mode": "affine"
837
+ },
838
+ "language_model.mtp.layers.0.self_attn.v_proj": {
839
+ "bits": 8,
840
+ "group_size": 128,
841
+ "mode": "affine"
842
+ }
843
+ },
844
+ "quantization_config": {
845
+ "bits": 8,
846
+ "group_size": 64,
847
+ "language_model.mtp.layers.0.mlp.shared_expert.down_proj": {
848
+ "bits": 8,
849
+ "group_size": 128,
850
+ "mode": "affine"
851
+ },
852
+ "language_model.mtp.layers.0.mlp.shared_expert.gate_proj": {
853
+ "bits": 8,
854
+ "group_size": 128,
855
+ "mode": "affine"
856
+ },
857
+ "language_model.mtp.layers.0.mlp.shared_expert.up_proj": {
858
+ "bits": 8,
859
+ "group_size": 128,
860
+ "mode": "affine"
861
+ },
862
+ "language_model.mtp.layers.0.self_attn.k_proj": {
863
+ "bits": 8,
864
+ "group_size": 128,
865
+ "mode": "affine"
866
+ },
867
+ "language_model.mtp.layers.0.self_attn.q_proj": {
868
+ "bits": 8,
869
+ "group_size": 128,
870
+ "mode": "affine"
871
+ },
872
+ "language_model.mtp.layers.0.self_attn.v_proj": {
873
+ "bits": 8,
874
+ "group_size": 128,
875
+ "mode": "affine"
876
+ }
877
+ }
878
+ },
879
+ "tie_word_embeddings": false,
880
+ "transformers_version": "5.2.0",
881
+ "video_token_id": 248057,
882
+ "vision_config": {
883
+ "deepstack_visual_indexes": [],
884
+ "depth": 27,
885
+ "dtype": "bfloat16",
886
+ "hidden_act": "gelu_pytorch_tanh",
887
+ "hidden_size": 1152,
888
+ "in_channels": 3,
889
+ "initializer_range": 0.02,
890
+ "intermediate_size": 4304,
891
+ "model_type": "qwen3_5_moe",
892
+ "num_heads": 16,
893
+ "num_position_embeddings": 2304,
894
+ "out_hidden_size": 2048,
895
+ "patch_size": 16,
896
+ "spatial_merge_size": 2,
897
+ "temporal_patch_size": 2
898
+ },
899
+ "vision_end_token_id": 248054,
900
+ "vision_start_token_id": 248053
901
+ }
figures/NEX_logo.svg ADDED
figures/Nex-N2-Benchmark-white.png ADDED

Git LFS Details

  • SHA256: 3fba33987e529cc9fffe491128389f7c162bab9b6cc0199cdab60c3c349a0a93
  • Pointer size: 131 Bytes
  • Size of remote file: 590 kB
model-00001-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:56f1d108197abf4a41a8da46340672a5697d95f7c7f4e5ec0e0d948833eb5ac6
3
+ size 5329916013
model-00002-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dce1ee4928dfbec2aef55509df9733524c5dc72cdf881b2a3f906423d73e3e5c
3
+ size 5365573922
model-00003-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c975b6e4dbefa731ee3ff0160420aaf920c5bf8bce01ff765a2ebf56a1090fef
3
+ size 5358684668
model-00004-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d6980cb10e61b10230089288050360ce650da8b8b4e874fe619c212165d791cb
3
+ size 5365574194
model-00005-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:72b9e730b455502ea3e4bdf493d4ab039d22c54c21f3033ecf1eb87e947f148e
3
+ size 5358684644
model-00006-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0b5880f39c1a8a3a468025dcf7d4763372b5006bcc9d4584eddc2bc4c21841e1
3
+ size 5365574174
model-00007-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8c9104dda8f6b3ef4585c227764e5b72a89b9ef853008e43382f8d90b68e09bd
3
+ size 5037050574
model-00008-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50cc1ff09601c52be4da9f5052f0c232d15b71c3c1579638ce84599fd2daa67e
3
+ size 540344667
model-mtp.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b355d970fe392a71c26e979766fcfff16f6fb2c9802bb15af73a92e937ef40d9
3
+ size 905113956
model.safetensors.index.json ADDED
The diff for this file is too large to render. See raw diff
 
preprocessor_config.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "size": {
3
+ "longest_edge": 16777216,
4
+ "shortest_edge": 65536
5
+ },
6
+ "patch_size": 16,
7
+ "temporal_patch_size": 2,
8
+ "merge_size": 2,
9
+ "image_mean": [
10
+ 0.5,
11
+ 0.5,
12
+ 0.5
13
+ ],
14
+ "image_std": [
15
+ 0.5,
16
+ 0.5,
17
+ 0.5
18
+ ],
19
+ "processor_class": "Qwen3VLProcessor",
20
+ "image_processor_type": "Qwen2VLImageProcessorFast"
21
+ }
processor_config.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "image_processor": {
3
+ "do_convert_rgb": true,
4
+ "do_normalize": true,
5
+ "do_rescale": true,
6
+ "image_mean": [
7
+ 0.5,
8
+ 0.5,
9
+ 0.5
10
+ ],
11
+ "image_processor_type": "Qwen3VLImageProcessor",
12
+ "image_std": [
13
+ 0.5,
14
+ 0.5,
15
+ 0.5
16
+ ],
17
+ "max_pixels": 16777216,
18
+ "merge_size": 2,
19
+ "min_pixels": 65536,
20
+ "patch_size": 16,
21
+ "rescale_factor": 0.00392156862745098,
22
+ "temporal_patch_size": 2
23
+ },
24
+ "processor_class": "Qwen3VLProcessor",
25
+ "video_processor": {
26
+ "do_convert_rgb": true,
27
+ "do_normalize": true,
28
+ "do_rescale": true,
29
+ "fps": 2.0,
30
+ "image_mean": [
31
+ 0.5,
32
+ 0.5,
33
+ 0.5
34
+ ],
35
+ "image_std": [
36
+ 0.5,
37
+ 0.5,
38
+ 0.5
39
+ ],
40
+ "max_frames": 768,
41
+ "max_pixels": 16777216,
42
+ "merge_size": 2,
43
+ "min_frames": 4,
44
+ "min_pixels": 65536,
45
+ "patch_size": 16,
46
+ "rescale_factor": 0.00392156862745098,
47
+ "temporal_patch_size": 2,
48
+ "video_processor_type": "Qwen3VLVideoProcessor"
49
+ }
50
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:87a7830d63fcf43bf241c3c5242e96e62dd3fdc29224ca26fed8ea333db72de4
3
+ size 19989343
tokenizer_config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "audio_bos_token": "<|audio_start|>",
4
+ "audio_eos_token": "<|audio_end|>",
5
+ "audio_token": "<|audio_pad|>",
6
+ "backend": "tokenizers",
7
+ "bos_token": null,
8
+ "clean_up_tokenization_spaces": false,
9
+ "eos_token": "<|im_end|>",
10
+ "errors": "replace",
11
+ "image_token": "<|image_pad|>",
12
+ "is_local": true,
13
+ "local_files_only": false,
14
+ "model_max_length": 262144,
15
+ "model_specific_special_tokens": {
16
+ "audio_bos_token": "<|audio_start|>",
17
+ "audio_eos_token": "<|audio_end|>",
18
+ "audio_token": "<|audio_pad|>",
19
+ "image_token": "<|image_pad|>",
20
+ "video_token": "<|video_pad|>",
21
+ "vision_bos_token": "<|vision_start|>",
22
+ "vision_eos_token": "<|vision_end|>"
23
+ },
24
+ "pad_token": "<|endoftext|>",
25
+ "pretokenize_regex": "(?i:'s|'t|'re|'ve|'m|'ll|'d)|[^\\r\\n\\p{L}\\p{N}]?[\\p{L}\\p{M}]+|\\p{N}| ?[^\\s\\p{L}\\p{M}\\p{N}]+[\\r\\n]*|\\s*[\\r\\n]+|\\s+(?!\\S)|\\s+",
26
+ "processor_class": "Qwen3VLProcessor",
27
+ "split_special_tokens": false,
28
+ "tokenizer_class": "TokenizersBackend",
29
+ "unk_token": null,
30
+ "video_token": "<|video_pad|>",
31
+ "vision_bos_token": "<|vision_start|>",
32
+ "vision_eos_token": "<|vision_end|>"
33
+ }