launch failed with EAGLE

#3
by luchangli03 - opened

Using this model, we can launch successfully without MTP, but when using MTP, crash occurs:
[2026-06-22 17:21:25 TP2] model.decoder.mlp.experts.w13_input_scale not found in params_dict.
[2026-06-22 17:21:25 TP2] model.decoder.mlp.experts.w13_input_scale not found in params_dict.
[2026-06-22 17:21:25 TP2] model.decoder.mlp.experts.w2_input_scale not found in params_dict.
[2026-06-22 17:21:25 TP2] model.decoder.mlp.experts.w13_input_scale not found in params_dict.
[2026-06-22 17:21:25 TP2] model.decoder.mlp.experts.w13_input_scale not found in params_dict.

[2026-06-22 17:21:25 TP1] Scheduler hit an exception: Traceback (most recent call last):
File "/sgl-workspace/sglang/python/sglang/srt/managers/scheduler.py", line 4060, in run_scheduler_process
scheduler = Scheduler(
^^^^^^^^^^
File "/sgl-workspace/sglang/python/sglang/srt/managers/scheduler.py", line 427, in init
self.init_model_worker()
File "/sgl-workspace/sglang/python/sglang/srt/managers/scheduler.py", line 818, in init_model_worker
self.maybe_init_draft_worker()
File "/sgl-workspace/sglang/python/sglang/srt/managers/scheduler.py", line 802, in maybe_init_draft_worker
self.draft_worker = DraftWorkerClass(**draft_worker_kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/sgl-workspace/sglang/python/sglang/srt/speculative/eagle_worker_v2.py", line 860, in init
self._draft_worker = EagleDraftWorker(
^^^^^^^^^^^^^^^^^
File "/sgl-workspace/sglang/python/sglang/srt/speculative/eagle_worker_v2.py", line 171, in init
self.draft_worker = TpModelWorker(
^^^^^^^^^^^^^^
File "/sgl-workspace/sglang/python/sglang/srt/managers/tp_worker.py", line 262, in init
self._init_model_runner()
File "/sgl-workspace/sglang/python/sglang/srt/managers/tp_worker.py", line 347, in _init_model_runner
self._model_runner = ModelRunner(
^^^^^^^^^^^^
File "/sgl-workspace/sglang/python/sglang/srt/model_executor/model_runner.py", line 571, in init
self.initialize(pre_model_load_memory)
File "/sgl-workspace/sglang/python/sglang/srt/model_executor/model_runner.py", line 679, in initialize
self.load_model()
File "/sgl-workspace/sglang/python/sglang/srt/model_executor/model_runner.py", line 1364, in load_model
self.model = self.loader.load_model(
^^^^^^^^^^^^^^^^^^^^^^^
File "/sgl-workspace/sglang/python/sglang/srt/model_loader/loader.py", line 2737, in load_model
return super().load_model(
^^^^^^^^^^^^^^^^^^^
File "/sgl-workspace/sglang/python/sglang/srt/model_loader/loader.py", line 743, in load_model
self.load_weights_and_postprocess(
File "/sgl-workspace/sglang/python/sglang/srt/model_loader/loader.py", line 763, in load_weights_and_postprocess
model.load_weights(weights)
File "/sgl-workspace/sglang/python/sglang/srt/models/deepseek_nextn.py", line 356, in load_weights
super().load_weights(weights, is_nextn=True)
File "/sgl-workspace/sglang/python/sglang/srt/models/deepseek_v2.py", line 2719, in load_weights
self.do_load_weights(weights, is_nextn)
File "/sgl-workspace/sglang/python/sglang/srt/models/deepseek_common/deepseek_weight_loader.py", line 373, in do_load_weights
future.result()
File "/usr/lib/python3.12/concurrent/futures/_base.py", line 449, in result
return self.__get_result()
^^^^^^^^^^^^^^^^^^^
File "/usr/lib/python3.12/concurrent/futures/_base.py", line 401, in __get_result
raise self._exception
File "/usr/lib/python3.12/concurrent/futures/thread.py", line 58, in run
result = self.fn(*self.args, **self.kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/sgl-workspace/sglang/python/sglang/srt/layers/moe/fused_moe_triton/layer.py", line 667, in weight_loader
self._weight_loader_physical(
File "/sgl-workspace/sglang/python/sglang/srt/layers/moe/fused_moe_triton/layer.py", line 697, in _weight_loader_physical
self._weight_loader_impl(
File "/sgl-workspace/sglang/python/sglang/srt/layers/moe/fused_moe_triton/layer.py", line 955, in _weight_loader_impl
self._load_model_weight_or_group_weight_scale(
File "/sgl-workspace/sglang/python/sglang/srt/layers/moe/fused_moe_triton/layer.py", line 399, in _load_model_weight_or_group_weight_scale
self._load_w13(
File "/sgl-workspace/sglang/python/sglang/srt/layers/moe/fused_moe_triton/layer.py", line 490, in load_w13
expert_data.copy
(loaded_weight)
RuntimeError: The size of tensor a (6144) must match the size of tensor b (3072) at non-singleton dimension 1

Worked for me with vLLM. seems like it is related to how sglang reads it?

Worked for me with tutelgroup/deepseek-671b

Sign up or log in to comment