sakamakismile commited on
Commit
d3272a8
·
verified ·
1 Parent(s): 39bdf30

Initial release: NVFP4 + MTP head bf16-restored, text-only

Browse files
.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,105 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: Qwen/Qwen3.6-27B
4
+ base_model_relation: quantized
5
+ library_name: transformers
6
+ tags:
7
+ - qwen3_5
8
+ - qwen3.6
9
+ - nvfp4
10
+ - quantized
11
+ - modelopt
12
+ - mtp
13
+ - speculative-decoding
14
+ - blackwell
15
+ - text-only
16
+ pipeline_tag: text-generation
17
+ language:
18
+ - en
19
+ - zh
20
+ - ja
21
+ - ko
22
+ - fr
23
+ - de
24
+ - es
25
+ - it
26
+ - pt
27
+ - ru
28
+ - ar
29
+ ---
30
+
31
+ # Qwen3.6-27B-Text-NVFP4-MTP
32
+
33
+ NVFP4-quantized text-only sibling of [`Qwen/Qwen3.6-27B`](https://huggingface.co/Qwen/Qwen3.6-27B), with the **MTP (Multi-Token Prediction) head restored in bf16** so speculative decoding actually works.
34
+
35
+ ## What's different from `sakamakismile/Qwen3.6-27B-NVFP4`
36
+
37
+ | | This repo (`-Text-NVFP4-MTP`) | `Qwen3.6-27B-NVFP4` |
38
+ |---|---|---|
39
+ | Quantization format | **`modelopt`** (vLLM SM120 native path) | `compressed-tensors` |
40
+ | MTP head | **Restored in bf16, working** | Dropped during export → 0% draft acceptance |
41
+ | Vision tower | **Stripped (text-only)** | Present (kept for VLM use) |
42
+ | Suggested launch | with `--speculative-config` | without speculation |
43
+
44
+ The original `Qwen3.6-27B-NVFP4` is left untouched so existing users (~15K downloads) are not disrupted. This is a focused text-only sibling for users who want maximum speed and don't need vision input.
45
+
46
+ ## Why this exists
47
+
48
+ Two HF Discussion threads on the original repo prompted this:
49
+
50
+ - [#5 — slower than official FP8 on Blackwell](https://huggingface.co/sakamakismile/Qwen3.6-27B-NVFP4/discussions/5) — root cause is the `compressed-tensors` NVFP4 path being slower than `modelopt` on Blackwell SM120; this repo uses `modelopt` natively.
51
+ - [#7 — MTP not responding](https://huggingface.co/sakamakismile/Qwen3.6-27B-NVFP4/discussions/7) — `AutoModelForCausalLM.from_pretrained` does not load the MTP head, so it gets dropped during quantization, leading to 0% draft acceptance. This repo grafts the 15 `mtp.*` tensors (bf16) back into the quantized checkpoint and adds them to the quantization ignore list.
52
+
53
+ Recipe is adapted from [`osoleve/Qwen3.5-27B-Text-NVFP4-MTP`](https://huggingface.co/osoleve/Qwen3.5-27B-Text-NVFP4-MTP) — credit and thanks.
54
+
55
+ ## Quantization details
56
+
57
+ - **Base**: `Qwen/Qwen3.6-27B` (bf16, 27.78B params, hybrid linear-attn + full-attn, 64 layers, 1 MTP layer)
58
+ - **Quantizer**: `nvidia-modelopt` 0.43.0 with `NVFP4_DEFAULT_CFG`
59
+ - **Calibration**: 20 samples from `neuralmagic/calibration` (LLM split), max_seq_len 8192
60
+ - **Ignored from quantization** (kept in bf16):
61
+ - `lm_head`
62
+ - All `model.visual.*` (vision tower) — then **physically deleted** in the text-only build
63
+ - All `*linear_attn.conv1d*` (Mamba-style SSM convolutions, 48 of the 64 layers)
64
+ - All `mtp.*` modules (the 1-layer MTP head: 15 tensors total, ~850 MB bf16)
65
+ - Other defaults from `NVFP4_DEFAULT_CFG`: `*router*`, `*mlp.gate.*`, `*block_sparse_moe.gate*`, `*output_layer*`
66
+
67
+ ## Usage with vLLM (Blackwell, SM120)
68
+
69
+ ### With MTP speculative decoding (recommended)
70
+
71
+ ```bash
72
+ vllm serve sakamakismile/Qwen3.6-27B-Text-NVFP4-MTP \
73
+ --trust-remote-code \
74
+ --gpu-memory-utilization 0.85 \
75
+ --max-model-len 8192 \
76
+ --quantization modelopt \
77
+ --speculative-config '{"method":"qwen3_5_mtp","num_speculative_tokens":1}'
78
+ ```
79
+
80
+ `num_speculative_tokens: 1` because the model has a single MTP layer (`mtp_num_hidden_layers=1`). The `qwen3_5_mtp` method handler is what vLLM uses for the Qwen3.5/3.6 family (internal `model_type: qwen3_5`); plain `mtp` also works as a fallback.
81
+
82
+ ### Without speculation
83
+
84
+ ```bash
85
+ vllm serve sakamakismile/Qwen3.6-27B-Text-NVFP4-MTP \
86
+ --trust-remote-code \
87
+ --gpu-memory-utilization 0.85 \
88
+ --max-model-len 8192 \
89
+ --quantization modelopt
90
+ ```
91
+
92
+ ## Hardware target
93
+
94
+ Built and tested on **NVIDIA RTX PRO 6000 Blackwell (SM120)**. Should also work on **RTX 5090** and other Blackwell consumer/workstation cards with sufficient VRAM (the model is roughly 14 GB after NVFP4 + ~850 MB of bf16 MTP/conv1d/lm_head).
95
+
96
+ ## Acknowledgements
97
+
98
+ - [`osoleve`](https://huggingface.co/osoleve) — for the MTP-restoration recipe on Qwen3.5
99
+ - [`Qwen`](https://huggingface.co/Qwen) — for the base model
100
+ - [`nvidia-modelopt`](https://github.com/NVIDIA/TensorRT-Model-Optimizer) team
101
+ - The reporters of Discussions #5 and #7 — for catching this cleanly
102
+
103
+ ## License
104
+
105
+ Apache 2.0 (inherited from `Qwen/Qwen3.6-27B`).
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
+ {{- raise_exception('System message must be at the beginning.') }}
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 (preserve_thinking is defined and preserve_thinking is true) or (loop.index0 > ns.last_query_index) %}
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 | string if args_value is string else args_value | tojson | safe %}
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>\n' }}
153
+ {%- endif %}
154
+ {%- endif %}
config.json ADDED
@@ -0,0 +1,215 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Qwen3_5ForConditionalGeneration"
4
+ ],
5
+ "dtype": "bfloat16",
6
+ "language_model_only": true,
7
+ "model_type": "qwen3_5",
8
+ "text_config": {
9
+ "attention_bias": false,
10
+ "attention_dropout": 0.0,
11
+ "attn_output_gate": true,
12
+ "bos_token_id": 248044,
13
+ "dtype": "bfloat16",
14
+ "eos_token_id": 248044,
15
+ "full_attention_interval": 4,
16
+ "head_dim": 256,
17
+ "hidden_act": "silu",
18
+ "hidden_size": 5120,
19
+ "initializer_range": 0.02,
20
+ "intermediate_size": 17408,
21
+ "layer_types": [
22
+ "linear_attention",
23
+ "linear_attention",
24
+ "linear_attention",
25
+ "full_attention",
26
+ "linear_attention",
27
+ "linear_attention",
28
+ "linear_attention",
29
+ "full_attention",
30
+ "linear_attention",
31
+ "linear_attention",
32
+ "linear_attention",
33
+ "full_attention",
34
+ "linear_attention",
35
+ "linear_attention",
36
+ "linear_attention",
37
+ "full_attention",
38
+ "linear_attention",
39
+ "linear_attention",
40
+ "linear_attention",
41
+ "full_attention",
42
+ "linear_attention",
43
+ "linear_attention",
44
+ "linear_attention",
45
+ "full_attention",
46
+ "linear_attention",
47
+ "linear_attention",
48
+ "linear_attention",
49
+ "full_attention",
50
+ "linear_attention",
51
+ "linear_attention",
52
+ "linear_attention",
53
+ "full_attention",
54
+ "linear_attention",
55
+ "linear_attention",
56
+ "linear_attention",
57
+ "full_attention",
58
+ "linear_attention",
59
+ "linear_attention",
60
+ "linear_attention",
61
+ "full_attention",
62
+ "linear_attention",
63
+ "linear_attention",
64
+ "linear_attention",
65
+ "full_attention",
66
+ "linear_attention",
67
+ "linear_attention",
68
+ "linear_attention",
69
+ "full_attention",
70
+ "linear_attention",
71
+ "linear_attention",
72
+ "linear_attention",
73
+ "full_attention",
74
+ "linear_attention",
75
+ "linear_attention",
76
+ "linear_attention",
77
+ "full_attention",
78
+ "linear_attention",
79
+ "linear_attention",
80
+ "linear_attention",
81
+ "full_attention",
82
+ "linear_attention",
83
+ "linear_attention",
84
+ "linear_attention",
85
+ "full_attention"
86
+ ],
87
+ "linear_conv_kernel_dim": 4,
88
+ "linear_key_head_dim": 128,
89
+ "linear_num_key_heads": 16,
90
+ "linear_num_value_heads": 48,
91
+ "linear_value_head_dim": 128,
92
+ "mamba_ssm_dtype": "float32",
93
+ "max_position_embeddings": 262144,
94
+ "model_type": "qwen3_5_text",
95
+ "mtp_num_hidden_layers": 1,
96
+ "mtp_use_dedicated_embeddings": false,
97
+ "num_attention_heads": 24,
98
+ "num_hidden_layers": 64,
99
+ "num_key_value_heads": 4,
100
+ "output_gate_type": "swish",
101
+ "pad_token_id": null,
102
+ "partial_rotary_factor": 0.25,
103
+ "rms_norm_eps": 1e-06,
104
+ "rope_parameters": {
105
+ "mrope_interleaved": true,
106
+ "mrope_section": [
107
+ 11,
108
+ 11,
109
+ 10
110
+ ],
111
+ "partial_rotary_factor": 0.25,
112
+ "rope_theta": 10000000,
113
+ "rope_type": "default"
114
+ },
115
+ "tie_word_embeddings": false,
116
+ "use_cache": true,
117
+ "vocab_size": 248320
118
+ },
119
+ "tie_word_embeddings": false,
120
+ "transformers_version": "5.5.4",
121
+ "quantization_config": {
122
+ "config_groups": {
123
+ "group_0": {
124
+ "input_activations": {
125
+ "dynamic": false,
126
+ "num_bits": 4,
127
+ "type": "float",
128
+ "group_size": 16
129
+ },
130
+ "weights": {
131
+ "dynamic": false,
132
+ "num_bits": 4,
133
+ "type": "float",
134
+ "group_size": 16
135
+ },
136
+ "targets": [
137
+ "Linear"
138
+ ]
139
+ }
140
+ },
141
+ "ignore": [
142
+ "lm_head",
143
+ "model.language_model.layers.0.linear_attn.conv1d",
144
+ "model.language_model.layers.1.linear_attn.conv1d",
145
+ "model.language_model.layers.10.linear_attn.conv1d",
146
+ "model.language_model.layers.12.linear_attn.conv1d",
147
+ "model.language_model.layers.13.linear_attn.conv1d",
148
+ "model.language_model.layers.14.linear_attn.conv1d",
149
+ "model.language_model.layers.16.linear_attn.conv1d",
150
+ "model.language_model.layers.17.linear_attn.conv1d",
151
+ "model.language_model.layers.18.linear_attn.conv1d",
152
+ "model.language_model.layers.2.linear_attn.conv1d",
153
+ "model.language_model.layers.20.linear_attn.conv1d",
154
+ "model.language_model.layers.21.linear_attn.conv1d",
155
+ "model.language_model.layers.22.linear_attn.conv1d",
156
+ "model.language_model.layers.24.linear_attn.conv1d",
157
+ "model.language_model.layers.25.linear_attn.conv1d",
158
+ "model.language_model.layers.26.linear_attn.conv1d",
159
+ "model.language_model.layers.28.linear_attn.conv1d",
160
+ "model.language_model.layers.29.linear_attn.conv1d",
161
+ "model.language_model.layers.30.linear_attn.conv1d",
162
+ "model.language_model.layers.32.linear_attn.conv1d",
163
+ "model.language_model.layers.33.linear_attn.conv1d",
164
+ "model.language_model.layers.34.linear_attn.conv1d",
165
+ "model.language_model.layers.36.linear_attn.conv1d",
166
+ "model.language_model.layers.37.linear_attn.conv1d",
167
+ "model.language_model.layers.38.linear_attn.conv1d",
168
+ "model.language_model.layers.4.linear_attn.conv1d",
169
+ "model.language_model.layers.40.linear_attn.conv1d",
170
+ "model.language_model.layers.41.linear_attn.conv1d",
171
+ "model.language_model.layers.42.linear_attn.conv1d",
172
+ "model.language_model.layers.44.linear_attn.conv1d",
173
+ "model.language_model.layers.45.linear_attn.conv1d",
174
+ "model.language_model.layers.46.linear_attn.conv1d",
175
+ "model.language_model.layers.48.linear_attn.conv1d",
176
+ "model.language_model.layers.49.linear_attn.conv1d",
177
+ "model.language_model.layers.5.linear_attn.conv1d",
178
+ "model.language_model.layers.50.linear_attn.conv1d",
179
+ "model.language_model.layers.52.linear_attn.conv1d",
180
+ "model.language_model.layers.53.linear_attn.conv1d",
181
+ "model.language_model.layers.54.linear_attn.conv1d",
182
+ "model.language_model.layers.56.linear_attn.conv1d",
183
+ "model.language_model.layers.57.linear_attn.conv1d",
184
+ "model.language_model.layers.58.linear_attn.conv1d",
185
+ "model.language_model.layers.6.linear_attn.conv1d",
186
+ "model.language_model.layers.60.linear_attn.conv1d",
187
+ "model.language_model.layers.61.linear_attn.conv1d",
188
+ "model.language_model.layers.62.linear_attn.conv1d",
189
+ "model.language_model.layers.8.linear_attn.conv1d",
190
+ "model.language_model.layers.9.linear_attn.conv1d",
191
+ "model.visual*",
192
+ "mtp.fc",
193
+ "mtp.layers.0.input_layernorm",
194
+ "mtp.layers.0.mlp.down_proj",
195
+ "mtp.layers.0.mlp.gate_proj",
196
+ "mtp.layers.0.mlp.up_proj",
197
+ "mtp.layers.0.post_attention_layernorm",
198
+ "mtp.layers.0.self_attn.k_norm",
199
+ "mtp.layers.0.self_attn.k_proj",
200
+ "mtp.layers.0.self_attn.o_proj",
201
+ "mtp.layers.0.self_attn.q_norm",
202
+ "mtp.layers.0.self_attn.q_proj",
203
+ "mtp.layers.0.self_attn.v_proj",
204
+ "mtp.norm",
205
+ "mtp.pre_fc_norm_embedding",
206
+ "mtp.pre_fc_norm_hidden"
207
+ ],
208
+ "quant_algo": "NVFP4",
209
+ "producer": {
210
+ "name": "modelopt",
211
+ "version": "0.43.0"
212
+ },
213
+ "quant_method": "modelopt"
214
+ }
215
+ }
generation_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 248044,
3
+ "do_sample": true,
4
+ "eos_token_id": [
5
+ 248046,
6
+ 248044
7
+ ],
8
+ "pad_token_id": 248044,
9
+ "temperature": 1.0,
10
+ "top_k": 20,
11
+ "top_p": 0.95,
12
+ "transformers_version": "5.5.4"
13
+ }
hf_quant_config.json ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "producer": {
3
+ "name": "modelopt",
4
+ "version": "0.43.0"
5
+ },
6
+ "quantization": {
7
+ "quant_algo": "NVFP4",
8
+ "kv_cache_quant_algo": null,
9
+ "group_size": 16,
10
+ "exclude_modules": [
11
+ "lm_head",
12
+ "model.language_model.layers.0.linear_attn.conv1d",
13
+ "model.language_model.layers.1.linear_attn.conv1d",
14
+ "model.language_model.layers.10.linear_attn.conv1d",
15
+ "model.language_model.layers.12.linear_attn.conv1d",
16
+ "model.language_model.layers.13.linear_attn.conv1d",
17
+ "model.language_model.layers.14.linear_attn.conv1d",
18
+ "model.language_model.layers.16.linear_attn.conv1d",
19
+ "model.language_model.layers.17.linear_attn.conv1d",
20
+ "model.language_model.layers.18.linear_attn.conv1d",
21
+ "model.language_model.layers.2.linear_attn.conv1d",
22
+ "model.language_model.layers.20.linear_attn.conv1d",
23
+ "model.language_model.layers.21.linear_attn.conv1d",
24
+ "model.language_model.layers.22.linear_attn.conv1d",
25
+ "model.language_model.layers.24.linear_attn.conv1d",
26
+ "model.language_model.layers.25.linear_attn.conv1d",
27
+ "model.language_model.layers.26.linear_attn.conv1d",
28
+ "model.language_model.layers.28.linear_attn.conv1d",
29
+ "model.language_model.layers.29.linear_attn.conv1d",
30
+ "model.language_model.layers.30.linear_attn.conv1d",
31
+ "model.language_model.layers.32.linear_attn.conv1d",
32
+ "model.language_model.layers.33.linear_attn.conv1d",
33
+ "model.language_model.layers.34.linear_attn.conv1d",
34
+ "model.language_model.layers.36.linear_attn.conv1d",
35
+ "model.language_model.layers.37.linear_attn.conv1d",
36
+ "model.language_model.layers.38.linear_attn.conv1d",
37
+ "model.language_model.layers.4.linear_attn.conv1d",
38
+ "model.language_model.layers.40.linear_attn.conv1d",
39
+ "model.language_model.layers.41.linear_attn.conv1d",
40
+ "model.language_model.layers.42.linear_attn.conv1d",
41
+ "model.language_model.layers.44.linear_attn.conv1d",
42
+ "model.language_model.layers.45.linear_attn.conv1d",
43
+ "model.language_model.layers.46.linear_attn.conv1d",
44
+ "model.language_model.layers.48.linear_attn.conv1d",
45
+ "model.language_model.layers.49.linear_attn.conv1d",
46
+ "model.language_model.layers.5.linear_attn.conv1d",
47
+ "model.language_model.layers.50.linear_attn.conv1d",
48
+ "model.language_model.layers.52.linear_attn.conv1d",
49
+ "model.language_model.layers.53.linear_attn.conv1d",
50
+ "model.language_model.layers.54.linear_attn.conv1d",
51
+ "model.language_model.layers.56.linear_attn.conv1d",
52
+ "model.language_model.layers.57.linear_attn.conv1d",
53
+ "model.language_model.layers.58.linear_attn.conv1d",
54
+ "model.language_model.layers.6.linear_attn.conv1d",
55
+ "model.language_model.layers.60.linear_attn.conv1d",
56
+ "model.language_model.layers.61.linear_attn.conv1d",
57
+ "model.language_model.layers.62.linear_attn.conv1d",
58
+ "model.language_model.layers.8.linear_attn.conv1d",
59
+ "model.language_model.layers.9.linear_attn.conv1d",
60
+ "model.visual*"
61
+ ]
62
+ }
63
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b2caa6a1d2c75662a98c081d9bc7f1c1f3471b6f7b1dad047afbd89acbfa5caa
3
+ size 19637775848
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,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ "model_max_length": 262144,
14
+ "model_specific_special_tokens": {
15
+ "audio_bos_token": "<|audio_start|>",
16
+ "audio_eos_token": "<|audio_end|>",
17
+ "audio_token": "<|audio_pad|>",
18
+ "image_token": "<|image_pad|>",
19
+ "video_token": "<|video_pad|>",
20
+ "vision_bos_token": "<|vision_start|>",
21
+ "vision_eos_token": "<|vision_end|>"
22
+ },
23
+ "pad_token": "<|endoftext|>",
24
+ "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+",
25
+ "split_special_tokens": false,
26
+ "tokenizer_class": "TokenizersBackend",
27
+ "unk_token": null,
28
+ "video_token": "<|video_pad|>",
29
+ "vision_bos_token": "<|vision_start|>",
30
+ "vision_eos_token": "<|vision_end|>"
31
+ }