Add coding-25-nonuniform

.gitattributes

@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 coding-25-uniform/tokenizer.json filter=lfs diff=lfs merge=lfs -text
+coding-25-nonuniform/tokenizer.json filter=lfs diff=lfs merge=lfs -text

coding-25-nonuniform/LOAD_VLLM.md

@@ -0,0 +1,43 @@
+# Loading this variant with vLLM
+
+The checkpoint has heterogeneous per-layer expert counts. vLLM's stock
+`Qwen3NextSparseMoeBlock` builds every layer with `config.num_experts`; our
+bundled `vllm_pruned_patch.py` overrides each layer to use its own count from
+`config.per_layer_num_experts`.
+
+## One-liner
+
+```bash
+PYTHONPATH=$(pwd):${PYTHONPATH:-} python -c "
+from vllm import LLM, SamplingParams
+llm = LLM(model='.', tensor_parallel_size=4, dtype='bfloat16',
+          gpu_memory_utilization=0.85, trust_remote_code=True,
+          enforce_eager=True)
+print(llm.generate(['def fib(n):'], SamplingParams(max_tokens=128))[0].outputs[0].text)
+"
+```
+
+## Why PYTHONPATH?
+
+vLLM spawns worker subprocesses via `multiprocessing.spawn` (safe with CUDA).
+Those workers re-import `vllm` fresh — any monkey-patch you applied in the
+parent process is gone. Python's `sitecustomize.py` mechanism runs
+automatically in **every** interpreter that has the relevant directory on
+`sys.path`. Putting this variant folder on `PYTHONPATH` is enough.
+
+## Tensor parallelism notes
+
+- **TP (tensor parallel)** works fine with heterogeneous counts — TP shards
+  hidden dimensions inside each expert.
+- **EP (expert parallel)** assumes experts shard evenly across ranks, which
+  is broken for heterogeneous counts. Keep `--enable-eplb` off.
+
+## With lm-eval-harness
+
+```bash
+PYTHONPATH=$(pwd):${PYTHONPATH:-} \
+  lm_eval --model vllm \
+    --model_args "pretrained=.,tensor_parallel_size=4,dtype=bfloat16,gpu_memory_utilization=0.85,max_model_len=4096,trust_remote_code=True,enforce_eager=True" \
+    --tasks humaneval,mbpp \
+    --batch_size auto
+```

coding-25-nonuniform/added_tokens.json

@@ -0,0 +1,28 @@
+{
+  "</think>": 151668,
+  "</tool_call>": 151658,
+  "</tool_response>": 151666,
+  "<think>": 151667,
+  "<tool_call>": 151657,
+  "<tool_response>": 151665,
+  "<|box_end|>": 151649,
+  "<|box_start|>": 151648,
+  "<|endoftext|>": 151643,
+  "<|file_sep|>": 151664,
+  "<|fim_middle|>": 151660,
+  "<|fim_pad|>": 151662,
+  "<|fim_prefix|>": 151659,
+  "<|fim_suffix|>": 151661,
+  "<|im_end|>": 151645,
+  "<|im_start|>": 151644,
+  "<|image_pad|>": 151655,
+  "<|object_ref_end|>": 151647,
+  "<|object_ref_start|>": 151646,
+  "<|quad_end|>": 151651,
+  "<|quad_start|>": 151650,
+  "<|repo_name|>": 151663,
+  "<|video_pad|>": 151656,
+  "<|vision_end|>": 151653,
+  "<|vision_pad|>": 151654,
+  "<|vision_start|>": 151652
+}

coding-25-nonuniform/chat_template.jinja

@@ -0,0 +1,117 @@
+{% macro render_extra_keys(json_dict, handled_keys) %}
+    {%- if json_dict is mapping %}
+        {%- for json_key in json_dict if json_key not in handled_keys %}
+            {%- if json_dict[json_key] is string %}
+                {{-'\n<' ~ json_key ~ '>' ~ (json_dict[json_key] | string) ~ '</' ~ json_key ~ '>' }}
+            {%- else %}
+                {{- '\n<' ~ json_key ~ '>' ~ (json_dict[json_key] | tojson | safe) ~ '</' ~ json_key ~ '>' }}
+            {%- endif %}
+        {%- endfor %}
+    {%- endif %}
+{%- endmacro %}
+
+{%- if messages[0]["role"] == "system" %}
+    {%- set system_message = messages[0]["content"] %}
+    {%- set loop_messages = messages[1:] %}
+{%- else %}
+    {%- set loop_messages = messages %}
+{%- endif %}
+
+{%- if not tools is defined %}
+    {%- set tools = [] %}
+{%- endif %}
+
+{%- if system_message is defined %}
+    {{- "<|im_start|>system\n" + system_message }}
+{%- else %}
+    {%- if tools is iterable and tools | length > 0 %}
+        {{- "<|im_start|>system\nYou are Qwen, a helpful AI assistant that can interact with a computer to solve tasks." }}
+    {%- endif %}
+{%- endif %}
+{%- if tools is iterable and tools | length > 0 %}
+    {{- "\n\n# Tools\n\nYou have access to the following functions:\n\n" }}
+    {{- "<tools>" }}
+    {%- for tool in tools %}
+        {%- if tool.function is defined %}
+            {%- set tool = tool.function %}
+        {%- endif %}
+        {{- "\n<function>\n<name>" ~ tool.name ~ "</name>" }}
+        {%- if tool.description is defined %}
+            {{- '\n<description>' ~ (tool.description | trim) ~ '</description>' }}
+        {%- endif %}
+        {{- '\n<parameters>' }}
+        {%- if tool.parameters is defined and tool.parameters is mapping and tool.parameters.properties is defined and tool.parameters.properties is mapping %}
+            {%- for param_name, param_fields in tool.parameters.properties|items %}
+                {{- '\n<parameter>' }}
+                {{- '\n<name>' ~ param_name ~ '</name>' }}
+                {%- if param_fields.type is defined %}
+                    {{- '\n<type>' ~ (param_fields.type | string) ~ '</type>' }}
+                {%- endif %}
+                {%- if param_fields.description is defined %}
+                    {{- '\n<description>' ~ (param_fields.description | trim) ~ '</description>' }}
+                {%- endif %}
+                {%- set handled_keys = ['name', 'type', 'description'] %}
+                {{- render_extra_keys(param_fields, handled_keys) }}
+                {{- '\n</parameter>' }}
+            {%- endfor %}
+        {%- endif %}
+        {%- set handled_keys = ['type', 'properties'] %}
+        {{- render_extra_keys(tool.parameters, handled_keys) }}
+        {{- '\n</parameters>' }}
+        {%- set handled_keys = ['type', 'name', 'description', 'parameters'] %}
+        {{- render_extra_keys(tool, handled_keys) }}
+        {{- '\n</function>' }}
+    {%- endfor %}
+    {{- "\n</tools>" }}
+    {{- '\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>' }}
+{%- endif %}
+{%- if system_message is defined %}
+    {{- '<|im_end|>\n' }}
+{%- else %}
+    {%- if tools is iterable and tools | length > 0 %}
+        {{- '<|im_end|>\n' }}
+    {%- endif %}
+{%- endif %}
+{%- for message in loop_messages %}
+    {%- if message.role == "assistant" and message.tool_calls is defined and message.tool_calls is iterable and message.tool_calls | length > 0 %}
+        {{- '<|im_start|>' + message.role }}
+        {%- if message.content is defined and message.content is string and message.content | trim | length > 0 %}
+            {{- '\n' + message.content | trim + '\n' }}
+        {%- endif %}
+        {%- for tool_call in message.tool_calls %}
+            {%- if tool_call.function is defined %}
+                {%- set tool_call = tool_call.function %}
+            {%- endif %}
+            {{- '\n<tool_call>\n<function=' + tool_call.name + '>\n' }}
+            {%- if tool_call.arguments is defined %}
+                {%- for args_name, args_value in tool_call.arguments|items %}
+                    {{- '<parameter=' + args_name + '>\n' }}
+                    {%- set args_value = args_value if args_value is string else args_value | tojson | safe %}
+                    {{- args_value }}
+                    {{- '\n</parameter>\n' }}
+                {%- endfor %}
+            {%- endif %}
+            {{- '</function>\n</tool_call>' }}
+        {%- endfor %}
+        {{- '<|im_end|>\n' }}
+    {%- elif message.role == "user" or message.role == "system" or message.role == "assistant" %}
+        {{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>' + '\n' }}
+    {%- elif message.role == "tool" %}
+        {%- if loop.previtem and loop.previtem.role != "tool" %}
+            {{- '<|im_start|>user' }}
+        {%- endif %}
+        {{- '\n<tool_response>\n' }}
+        {{- message.content }}
+        {{- '\n</tool_response>' }}
+        {%- if not loop.last and loop.nextitem.role != "tool" %}
+            {{- '<|im_end|>\n' }}
+        {%- elif loop.last %}
+            {{- '<|im_end|>\n' }}
+        {%- endif %}
+    {%- else %}
+        {{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>\n' }}
+    {%- endif %}
+{%- endfor %}
+{%- if add_generation_prompt %}
+    {{- '<|im_start|>assistant\n' }}
+{%- endif %}

coding-25-nonuniform/config.json

@@ -0,0 +1,145 @@
+{
+  "architectures": [
+    "Qwen3NextForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0,
+  "bos_token_id": 151643,
+  "decoder_sparse_step": 1,
+  "dtype": "bfloat16",
+  "eos_token_id": 151645,
+  "full_attention_interval": 4,
+  "head_dim": 256,
+  "hidden_act": "silu",
+  "hidden_size": 2048,
+  "initializer_range": 0.02,
+  "intermediate_size": 5120,
+  "layer_types": [
+    "linear_attention",
+    "linear_attention",
+    "linear_attention",
+    "full_attention",
+    "linear_attention",
+    "linear_attention",
+    "linear_attention",
+    "full_attention",
+    "linear_attention",
+    "linear_attention",
+    "linear_attention",
+    "full_attention",
+    "linear_attention",
+    "linear_attention",
+    "linear_attention",
+    "full_attention",
+    "linear_attention",
+    "linear_attention",
+    "linear_attention",
+    "full_attention",
+    "linear_attention",
+    "linear_attention",
+    "linear_attention",
+    "full_attention",
+    "linear_attention",
+    "linear_attention",
+    "linear_attention",
+    "full_attention",
+    "linear_attention",
+    "linear_attention",
+    "linear_attention",
+    "full_attention",
+    "linear_attention",
+    "linear_attention",
+    "linear_attention",
+    "full_attention",
+    "linear_attention",
+    "linear_attention",
+    "linear_attention",
+    "full_attention",
+    "linear_attention",
+    "linear_attention",
+    "linear_attention",
+    "full_attention",
+    "linear_attention",
+    "linear_attention",
+    "linear_attention",
+    "full_attention"
+  ],
+  "linear_conv_kernel_dim": 4,
+  "linear_key_head_dim": 128,
+  "linear_num_key_heads": 16,
+  "linear_num_value_heads": 32,
+  "linear_value_head_dim": 128,
+  "max_position_embeddings": 262144,
+  "mlp_only_layers": [],
+  "model_type": "qwen3_next",
+  "moe_intermediate_size": 512,
+  "norm_topk_prob": true,
+  "num_attention_heads": 16,
+  "num_experts": 512,
+  "num_experts_per_tok": 10,
+  "num_hidden_layers": 48,
+  "num_key_value_heads": 2,
+  "original_num_experts": 512,
+  "output_router_logits": false,
+  "partial_rotary_factor": 0.25,
+  "per_layer_num_experts": [
+    453,
+    398,
+    460,
+    357,
+    354,
+    405,
+    439,
+    429,
+    380,
+    359,
+    342,
+    370,
+    338,
+    351,
+    379,
+    407,
+    373,
+    416,
+    439,
+    432,
+    399,
+    363,
+    361,
+    395,
+    352,
+    370,
+    409,
+    415,
+    394,
+    416,
+    439,
+    417,
+    382,
+    351,
+    354,
+    390,
+    350,
+    354,
+    374,
+    403,
+    404,
+    390,
+    359,
+    353,
+    355,
+    331,
+    320,
+    351
+  ],
+  "rms_norm_eps": 1e-06,
+  "rope_scaling": null,
+  "rope_theta": 5000000,
+  "router_aux_loss_coef": 0.001,
+  "shared_expert_intermediate_size": 512,
+  "tie_word_embeddings": false,
+  "transformers_version": "4.57.6",
+  "use_cache": true,
+  "use_sliding_window": false,
+  "vocab_size": 151936
+}

coding-25-nonuniform/generation_config.json

@@ -0,0 +1,12 @@
+{
+  "bos_token_id": 151643,
+  "do_sample": true,
+  "eos_token_id": [
+    151645,
+    151643
+  ],
+  "pad_token_id": 151643,
+  "top_k": 40,
+  "top_p": 0.95,
+  "transformers_version": "4.57.6"
+}

coding-25-nonuniform/model-00001-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8eec7c9f66e76d1e48b083c356844841a69bf2d2d31f5ddbd1621d5062e0b8ac
+size 4999298520

coding-25-nonuniform/model-00002-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3e4147f5fa966b5cad069b781b195a8d35e72db7a3141524c180aa6cfd53c55c
+size 4999405848

coding-25-nonuniform/model-00003-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8c28f1e8c3f4d8bc3e8a21f4b41015052aa2f2496cc992bf5ad09fb5ab2daa11
+size 4999494688

coding-25-nonuniform/model-00004-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:276aaf64663bb98bb45c49894ff54159092495e7d3bfbaad9b0278521fb42240
+size 4999614736

coding-25-nonuniform/model-00005-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d0a52710157bf8cfcbeb53ef706a1f0bb0e89a09ba63e488a8978686d8fc309a
+size 4999412816

coding-25-nonuniform/model-00006-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e16f947b7958f2ba3b780b27ecdeab8cecb99ccd6fb376856f622272ee61f45f
+size 4998977312

coding-25-nonuniform/model-00007-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fcc24fff499c591b3f2bf6ab75f39bc767fb9029556d024e08bc3b38f1445dc5
+size 4999210312

coding-25-nonuniform/model-00008-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b58b74dfda4ee121239b99f73b9b7f801b4ca9f263117480e9c4504c58b8823f
+size 4999281256

coding-25-nonuniform/model-00009-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:502fce4780567ba8083b2631d8801b78dc33d1ce9dfeb6749223e9ae389294c4
+size 4999619952

coding-25-nonuniform/model-00010-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8caddafc0e4817d40ed47c3a9579c77188ab05eff81b10b2cb621374f1c77cc6
+size 4999629376

coding-25-nonuniform/model-00011-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9f4c610399326faf124800320be41b25cbdddfffe81d9cb3050157f4cce831ee
+size 4999509320

coding-25-nonuniform/model-00012-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4b292db2baad3dc6b7b7abfd9a2d92a54f49186efbacc6f601816f1329c368a8
+size 4999158336

coding-25-nonuniform/model-00013-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bbddb8593f582ae6f0b7c2491bf34003ad88a4114f0db0ea01e38385bcd1f7d1
+size 4999345480

coding-25-nonuniform/model-00014-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b96b3d684d67233dbf164408d0005d580144cd5110f61d48bb38d6c5fe697af5
+size 4999436864

coding-25-nonuniform/model-00015-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:03768828230d921db71aece80dea42790340405e69f29dac9f5fee105fe54e4e
+size 4999705928

coding-25-nonuniform/model-00016-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c8bf8b1fbc6ce5427710ac6b67f24d123d468c83789c696cad64c0b42f7e9732
+size 4999568072

coding-25-nonuniform/model-00017-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ab4ad44033d31a2a3e524bdaf3095f81a82b019bf34871bf4aa192336f25317f
+size 4999390616

coding-25-nonuniform/model-00018-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d1e919762a2c321fa8881caac29e513fa33f5171882675309838d48d49caf528
+size 4999108976

coding-25-nonuniform/model-00019-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a1ed2f79f6301abaa361f1b1d0dd9f2c0480f93fabffcfae353206b6dd55f0d0
+size 4999271728

coding-25-nonuniform/model-00020-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:83100657e186e0acbf1036da86cc4af684475f7e3b12132fa137a639c685e0be
+size 4999244352

coding-25-nonuniform/model-00021-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1f36750e050454921bc5f568ea959d81db39a5c1e225e244a8f6e5fd2a64ac30
+size 4999640392

coding-25-nonuniform/model-00022-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c6580ce45968f89cc1a6ba4e6f06fa2f53a46efbf74df3093e0b9e7734642b6d
+size 4998978112

coding-25-nonuniform/model-00023-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0ef5f8bdbd85f3fccf0371f78ed2859a99921fc21978b8f29d20b6c5923555b0
+size 4999198024

coding-25-nonuniform/model-00024-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:99afbca833886c070d9184342e9cb220a655d4270f2f3345ba0e14ac12bcf47c
+size 4998810216

coding-25-nonuniform/model-00025-of-00025.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d818d704bc49fda52b3fa8de4686b1095dacfce500a33eb7a710614551a5c3c6
+size 691556896

coding-25-nonuniform/pruned_metadata.json

@@ -0,0 +1,60 @@
+{
+  "base_model": "Qwen/Qwen3-Coder-Next",
+  "prune_mask": "/nfs/scistore19/alistgrp/mhelcig/repos/rco/llm/results/coder_next/mask_perexpert_reap_25pct.pt",
+  "original_num_experts": 512,
+  "num_hidden_layers": 48,
+  "num_experts_per_tok": 10,
+  "per_layer_num_experts": [
+    453,
+    398,
+    460,
+    357,
+    354,
+    405,
+    439,
+    429,
+    380,
+    359,
+    342,
+    370,
+    338,
+    351,
+    379,
+    407,
+    373,
+    416,
+    439,
+    432,
+    399,
+    363,
+    361,
+    395,
+    352,
+    370,
+    409,
+    415,
+    394,
+    416,
+    439,
+    417,
+    382,
+    351,
+    354,
+    390,
+    350,
+    354,
+    374,
+    403,
+    404,
+    390,
+    359,
+    353,
+    355,
+    331,
+    320,
+    351
+  ],
+  "total_pruned": 6144,
+  "sparsity": 0.25,
+  "allocation": "nonuniform"
+}

coding-25-nonuniform/sitecustomize.py

@@ -0,0 +1,12 @@
+"""Applied automatically in every Python process on PYTHONPATH.
+vLLM spawns worker subprocesses that re-import modules; without this file
+the monkey-patch would be missing in workers and weight loading would fail
+for the heterogeneous per-layer expert counts.
+"""
+import os
+if os.environ.get('VLLM_PRUNED_PATCH_DISABLE') != '1':
+    try:
+        import vllm_pruned_patch
+        vllm_pruned_patch.apply()
+    except Exception:
+        pass

coding-25-nonuniform/special_tokens_map.json

@@ -0,0 +1,31 @@
+{
+  "additional_special_tokens": [
+    "<|im_start|>",
+    "<|im_end|>",
+    "<|object_ref_start|>",
+    "<|object_ref_end|>",
+    "<|box_start|>",
+    "<|box_end|>",
+    "<|quad_start|>",
+    "<|quad_end|>",
+    "<|vision_start|>",
+    "<|vision_end|>",
+    "<|vision_pad|>",
+    "<|image_pad|>",
+    "<|video_pad|>"
+  ],
+  "eos_token": {
+    "content": "<|im_end|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

coding-25-nonuniform/tokenizer.json

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aeb13307a71acd8fe81861d94ad54ab689df773318809eed3cbe794b4492dae4
+size 11422654

coding-25-nonuniform/tokenizer_config.json

@@ -0,0 +1,239 @@
+{
+  "add_bos_token": false,
+  "add_prefix_space": false,
+  "added_tokens_decoder": {
+    "151643": {
+      "content": "<|endoftext|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151644": {
+      "content": "<|im_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151645": {
+      "content": "<|im_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151646": {
+      "content": "<|object_ref_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151647": {
+      "content": "<|object_ref_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151648": {
+      "content": "<|box_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151649": {
+      "content": "<|box_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151650": {
+      "content": "<|quad_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151651": {
+      "content": "<|quad_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151652": {
+      "content": "<|vision_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151653": {
+      "content": "<|vision_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151654": {
+      "content": "<|vision_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151655": {
+      "content": "<|image_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151656": {
+      "content": "<|video_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151657": {
+      "content": "<tool_call>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151658": {
+      "content": "</tool_call>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151659": {
+      "content": "<|fim_prefix|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151660": {
+      "content": "<|fim_middle|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151661": {
+      "content": "<|fim_suffix|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151662": {
+      "content": "<|fim_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151663": {
+      "content": "<|repo_name|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151664": {
+      "content": "<|file_sep|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151665": {
+      "content": "<tool_response>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151666": {
+      "content": "</tool_response>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151667": {
+      "content": "<think>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151668": {
+      "content": "</think>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    }
+  },
+  "additional_special_tokens": [
+    "<|im_start|>",
+    "<|im_end|>",
+    "<|object_ref_start|>",
+    "<|object_ref_end|>",
+    "<|box_start|>",
+    "<|box_end|>",
+    "<|quad_start|>",
+    "<|quad_end|>",
+    "<|vision_start|>",
+    "<|vision_end|>",
+    "<|vision_pad|>",
+    "<|image_pad|>",
+    "<|video_pad|>"
+  ],
+  "bos_token": null,
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "<|im_end|>",
+  "errors": "replace",
+  "extra_special_tokens": {},
+  "model_max_length": 1048576,
+  "pad_token": "<|endoftext|>",
+  "split_special_tokens": false,
+  "tokenizer_class": "Qwen2Tokenizer",
+  "unk_token": null
+}

coding-25-nonuniform/vllm_pruned_patch.py

@@ -0,0 +1,88 @@
+"""Monkey-patch vLLM's Qwen3-Next MoE block to support heterogeneous per-layer expert counts.
+
+Use:
+    import vllm_pruned_patch
+    vllm_pruned_patch.apply()
+    # ...then any vLLM import / engine / lcb_runner call
+
+The patch reads `config.per_layer_num_experts` (a list of length num_hidden_layers)
+written by `prune_and_save.py`. If that field is missing the patch is a no-op,
+so it's safe to apply unconditionally.
+
+Tested against vLLM 0.16.0 / Qwen3NextSparseMoeBlock.
+EP (expert parallelism) is NOT supported with heterogeneous counts —
+keep --enable-eplb off (the default) and stick to TP-only.
+"""
+
+from __future__ import annotations
+
+import logging
+
+logger = logging.getLogger(__name__)
+
+_PATCHED = False
+
+
+def apply():
+    """Idempotently install the patch. Safe to call multiple times."""
+    global _PATCHED
+    if _PATCHED:
+        return
+
+    from vllm.model_executor.models import qwen3_next as qm
+    from vllm.model_executor.models.utils import extract_layer_index
+
+    OriginalBlock = qm.Qwen3NextSparseMoeBlock
+    original_init = OriginalBlock.__init__
+
+    def patched_init(self, vllm_config, prefix: str = ""):
+        cfg = vllm_config.model_config.hf_config
+        per_layer = getattr(cfg, "per_layer_num_experts", None)
+        if per_layer is None:
+            # No pruning metadata — fall through to original behavior.
+            return original_init(self, vllm_config, prefix=prefix)
+
+        layer_idx = extract_layer_index(prefix)
+        n_local = int(per_layer[layer_idx])
+        original_n = cfg.num_experts
+
+        # Temporarily override num_experts on the (shared) hf_config so the
+        # original __init__ constructs gate + FusedMoE with the right size.
+        # Restore immediately after to avoid bleeding into other layers.
+        try:
+            cfg.num_experts = n_local
+            original_init(self, vllm_config, prefix=prefix)
+        finally:
+            cfg.num_experts = original_n
+
+    qm.Qwen3NextSparseMoeBlock.__init__ = patched_init
+
+    # Patch get_expert_mapping to use max kept count (smaller mapping table).
+    OriginalLM = qm.Qwen3NextForCausalLM
+    original_get_expert_mapping = OriginalLM.get_expert_mapping
+
+    def patched_get_expert_mapping(self):
+        per_layer = getattr(self.config, "per_layer_num_experts", None)
+        if per_layer is None:
+            return original_get_expert_mapping(self)
+        from vllm.model_executor.layers.fused_moe.shared_fused_moe import (
+            SharedFusedMoE,
+        )
+        # Saved checkpoint always has expert ids 0..max-1 across layers
+        # (each layer is re-indexed densely in prune_and_save.py).
+        max_n = max(int(x) for x in per_layer)
+        return SharedFusedMoE.make_expert_params_mapping(
+            self,
+            ckpt_gate_proj_name="gate_proj",
+            ckpt_down_proj_name="down_proj",
+            ckpt_up_proj_name="up_proj",
+            num_experts=max_n,
+            num_redundant_experts=self.num_redundant_experts,
+        )
+
+    qm.Qwen3NextForCausalLM.get_expert_mapping = patched_get_expert_mapping
+
+    _PATCHED = True
+    logger.info(
+        "Applied Qwen3-Next pruned-model patch (heterogeneous per-layer expert counts)."
+    )