diff --git a/python/sglang/srt/model_executor/runner/eager_runner.py b/python/sglang/srt/model_executor/runner/eager_runner.py index a7a76c2ed..7d10c2846 100644 --- a/python/sglang/srt/model_executor/runner/eager_runner.py +++ b/python/sglang/srt/model_executor/runner/eager_runner.py @@ -317,6 +317,7 @@ class EagerRunner(BaseRunner): model_runner.moe_layers, model_runner.moe_fusions, dsa_indexers=model_runner.dsa_indexers, + mha_companion_layers=model_runner.mha_companion_layers, ), ): ret = model_runner.model.forward(