Skip to content

Support quantized DSpark Markov heads - #50424

Merged
vllm-bot merged 4 commits into
vllm-project:mainfrom
askliar:feat/add_support_for_quantized_markov_head
Aug 3, 2026
Merged

vllm-bot merged 4 commits into
vllm-project:mainfrom
askliar:feat/add_support_for_quantized_markov_head

Conversation

@askliar

@askliar askliar commented Jul 30, 2026

Copy link
Copy Markdown
Contributor

Summary

Add quantization configuration support to the DSpark Markov head.

DSparkMarkovHead now accepts and forwards quant_config to its ParallelLMHead-based markov_w2 projection. Qwen3DSparkModel passes its model quantization configuration when constructing the Markov head.

This allows quantized markov_w2 weights, including W4A16 configurations with weight_scale_2, to load through the normal quantization dispatch path while preserving unquantized behavior.

Testing

Attempted DSpark quantization and loading with this branch.

@claude claude Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Claude Code Review

This pull request is from a fork — automated review is disabled. A repository maintainer can comment @claude review to run a one-time review.

@mergify mergify Bot added quantization qwen Related to Qwen models labels Jul 30, 2026
@benchislett benchislett added ready ONLY add when PR is ready to merge/full CI is needed verified Run pre-commit for new contributors without triggering other tests labels Jul 30, 2026
@benchislett
benchislett enabled auto-merge (squash) July 30, 2026 16:49
Andrii Skliar added 2 commits August 2, 2026 14:58
…SparkModel

Signed-off-by: Andrii Skliar <askliar@nvidia.com>
Signed-off-by: Andrii Skliar <askliar@nvidia.com>
auto-merge was automatically disabled August 2, 2026 14:59

Head branch was pushed to by a user without write access

@askliar
askliar force-pushed the feat/add_support_for_quantized_markov_head branch from 2250f9c to 5dbdce4 Compare August 2, 2026 14:59
@benchislett
benchislett enabled auto-merge (squash) August 2, 2026 15:09
@vllm-bot
vllm-bot merged commit b977407 into vllm-project:main Aug 3, 2026
88 of 91 checks passed
Mazyod pushed a commit to Mazyod/vllm that referenced this pull request Aug 11, 2026
Signed-off-by: Andrii Skliar <askliar@nvidia.com>
Co-authored-by: Andrii Skliar <askliar@nvidia.com>
Co-authored-by: Benjamin Chislett <bchislett@nvidia.com>
(cherry picked from commit b977407)
D-G-Dimitrov added a commit to D-G-Dimitrov/vllm that referenced this pull request Aug 26, 2026
…alError)

The rebase onto v0.27.1 merged upstream vllm-project#50424's quant_config into
DSparkMarkovHead.__init__ but dropped the fork's keyword-only
shard_vocab: bool | None = None parameter during conflict resolution,
while keeping the body hunk that reads it. At model load every DSpark
draft model (DSv4-NVIDIA/Qwen3) hits 'if shard_vocab is None:' with
shard_vocab never bound -> UnboundLocalError at qwen3_dspark.py:89.

Restore the keyword-only shard_vocab parameter after quant_config;
all callers pass their args by keyword (prod and tests), so the
signature stays backward compatible.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

quantization qwen Related to Qwen models ready ONLY add when PR is ready to merge/full CI is needed verified Run pre-commit for new contributors without triggering other tests

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants