Skip to content

Clear Qwen text rope state before VLM prefill#333

Merged
neilmehta24 merged 1 commit into
mainfrom
neil/qwen-rope-state
Jun 3, 2026
Merged

Clear Qwen text rope state before VLM prefill#333
neilmehta24 merged 1 commit into
mainfrom
neil/qwen-rope-state

Conversation

@neilmehta24

Copy link
Copy Markdown
Member

Fixes #331

When a text-only prompt is added to the Qwen batch generator, if there are active decoding sequences containing images, the MRoPE state will not be correct for the text-only prefill.

So, clear the MRoPE state before computing the text-only prefill.

@github-actions github-actions Bot added the CLA signed Indicates that all contributors have signed label Jun 2, 2026
@neilmehta24 neilmehta24 marked this pull request as ready for review June 2, 2026 20:11
@neilmehta24 neilmehta24 merged commit e47768b into main Jun 3, 2026
2 checks passed
@neilmehta24 neilmehta24 deleted the neil/qwen-rope-state branch June 3, 2026 17:08
@github-actions github-actions Bot locked and limited conversation to collaborators Jun 3, 2026
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

CLA signed Indicates that all contributors have signed

Projects

None yet

Development

Successfully merging this pull request may close these issues.

MLX Engine _rope_deltas Shape Mismatch Causes Model Crash

2 participants