[hotfix] Incorrect CombineOverlapArgs in SBO (#12230)

2025-10-28 01:23:06 -07:00
parent cc7b04a29c
commit b1e13e7cea
3 changed files with 16 additions and 11 deletions
@@ -98,7 +98,10 @@ def execute_sbo(
        ):
            forward_shared_experts()

-    hidden_states = experts.dispatcher.combine(combine_input=combine_input)
+    hidden_states = experts.dispatcher.combine(
+        combine_input=combine_input,
+        overlap_args=combine_overlap_args,
+    )

    return hidden_states