Repository navigation
Conversation
Add set_eagle3_layers_to_capture method to Qwen3_5ForCausalLM so that the _is_layer_to_capture attribute is correctly set on captured layers when called from Qwen3VLForConditionalGeneration, instead of only assigning the list without marking the layers.
|
Warning You have reached your daily quota limit. Please wait up to 24 hours and I will start processing your requests again! |
MotivationWhen running Qwen3.5 MoE with EAGLE3 using the latest SGLang code, server startup fails because the target model does not correctly expose the EAGLE3 layer-capture setup path. The observed error is: Qwen3.5 MoE + EAGLE3 needs to capture auxiliary hidden states from selected target-model layers. Without properly setting the captured layers, the EAGLE3 draft-and-verify path cannot be initialized correctly. This PR fixes the layer-capture setup for Qwen3.5 by ensuring that captured layers are marked with With this fix, Qwen3.5 MoE + EAGLE3 can successfully start the SGLang server and run inference. The public EAGLE3 draft model used for validation is available at: |
|
Thanks @jiahang01. Closing this because it has had no updates in 91 days. Reopen it if the work is still relevant. Some directories moved recently, so an older branch may need retargeting: |
* chore(build): patch sgl-project/sglang#25408 * chore(build): patch sgl-project/sglang#35371
Motivation
Qwen3.5 VL/MoE with EAGLE3 requires auxiliary hidden states from selected target model layers. The wrapper calls
set_eagle3_layers_to_capture, but Qwen3.5 only assignedlayers_to_capturewithout marking the actual decoder layers with_is_layer_to_capture.As a result, aux hidden states were not captured correctly and EAGLE3 execution could fail when starting Qwen3.5 MoE models.
Modifications
set_eagle3_layers_to_capturetoQwen3_5ForCausalLM._is_layer_to_capture, matching the existing DFLASH capture path.Qwen3VLForConditionalGeneration.set_eagle3_layers_to_captureto call the language model capture method when available, instead of only assigninglayers_to_capture.Accuracy Tests
Manually verified that Qwen3.5 MoE with EAGLE3 can start and run with this patch.
The EAGLE3 draft model used for validation is publicly available at:
https://huggingface.co/AQ-MedAI/Qwen3.5-397B-A17B-eagle3
Speed Tests and Profiling
Not applicable. This change only fixes layer capture setup for EAGLE3 auxiliary hidden states and does not change kernels or inference computation paths outside the capture configuration.
Checklist