[XPU][UT]Fix InternS1ProForConditionalGeneration AssertionError (#48232)
Signed-off-by: Lai, Yejing <yejing.lai@intel.com>
Yejing Lai committed
Jul 11, 2026 at 21:56 UTC
76fedaa2a5382e4b76d236253da55b0646b1459c
1 file changed
+6
tests/models/utils.py
+6
@@ -468,6 +468,9 @@ def dummy_hf_overrides(
468
# Kimi uses `num_expert_group` instead of `n_group`.
469
if n_group is None:
470
n_group = getattr(text_config, "num_expert_group", None)
471
+ # InternS1Pro uses `router_n_groups` instead of `n_group`.
472
+ if n_group is None:
473
+ n_group = getattr(text_config, "router_n_groups", None)
474
num_experts = n_group * 2 if n_group is not None else 2
475
476
# we use three layers for Gemma-3n to check
@@ -515,6 +518,9 @@ def dummy_hf_overrides(
518
"EagleLlama4ForCausalLM",
519
):
520
num_experts_per_tok = 1
521
+ elif model_arch == "InternS1ProForConditionalGeneration":
522
+ assert n_group is not None
523
+ num_experts_per_tok = n_group
524
update_dict.update(
525
{
526
"num_experts": num_experts,