Support DSpark configs with `architectures=DSparkDraftModel` + `model_type=qwen3` (#52197)
Signed-off-by: mgoin <mgoin64@gmail.com>
Misha Goin committed
Aug 17, 2026 at 12:48 UTC
7075ddac28c25d4fd2b84bc2a9a6c5ffde0345c8
1 file changed
+14
-1
vllm/config/speculative.py
+14
-1
@@ -953,6 +953,10 @@ class SpeculativeConfig:
953
"dspark" in self.draft_model_config.model.lower()
954
or "Qwen3DSparkModel" in self.draft_model_config.architectures
955
or "Gemma4DSparkModel" in self.draft_model_config.architectures
956
+ or (
957
+ "DSparkDraftModel" in self.draft_model_config.architectures
958
+ and self.draft_model_config.hf_config.model_type == "qwen3"
959
+ )
960
):
961
self.method = "dspark"
962
elif self.draft_model_config.hf_config.model_type == "medusa":
@@ -1010,7 +1014,16 @@ class SpeculativeConfig:
1014
self.draft_model_config.hf_config = eagle_config
1015
self.update_arch_()
1016
1013
- if self.method == "dspark" and (
1017
+ if (
1018
+ self.method == "dspark"
1019
+ and "DSparkDraftModel" in self.draft_model_config.architectures
1020
+ and self.draft_model_config.hf_config.model_type == "qwen3"
1021
+ ):
1022
+ self.draft_model_config.hf_config.architectures = [
1023
+ "Qwen3DSparkModel"
1024
+ ]
1025
+ self.update_arch_()
1026
+ elif self.method == "dspark" and (
1027
"Qwen3DSparkModel" not in self.draft_model_config.architectures
1028
and "Gemma4DSparkModel" not in self.draft_model_config.architectures
1029
and "K3DSparkModel" not in self.draft_model_config.architectures