Update get_expert_linear_names

ajrasane · ajrasane · commit 9e3b3991c0a2 · 2026-04-01T22:08:42.000Z
Signed-off-by: ajrasane &lt;131806219+ajrasane@users.noreply.github.com&gt;
diff --git a/examples/llm_ptq/run_vllm.py b/examples/llm_ptq/run_vllm.py
@@ -116,6 +116,7 @@ def main():
         max_model_len=args.max_model_len,
         trust_remote_code=args.trust_remote_code,
         quantization=quantization,
+        enforce_eager=True,
     )
 
     # Get sampling params from config, with CLI/defaults as fallback
diff --git a/modelopt/torch/export/layer_utils.py b/modelopt/torch/export/layer_utils.py
@@ -972,6 +972,7 @@ def module_match_name_list(module, name_list):
             "Qwen3MoeSparseMoeBlock",
             "Qwen3NextSparseMoeBlock",
             "Qwen3_5MoeSparseMoeBlock",
+            "Qwen3OmniMoeThinkerTextSparseMoeBlock",
             "DeepseekMoE",
         ],
     ):

Original file line number	Diff line number	Diff line change
`@@ -116,6 +116,7 @@ def main():`
`116`	`116`	`max_model_len=args.max_model_len,`
`117`	`117`	`trust_remote_code=args.trust_remote_code,`
`118`	`118`	`quantization=quantization,`
	`119`	`+ enforce_eager=True,`
`119`	`120`	`)`
`120`	`121`
`121`	`122`	`# Get sampling params from config, with CLI/defaults as fallback`