Skip to content

[Bugfix] Fix mm budget setting for Qwen Omni models#33634

Merged
DarkLight1337 merged 2 commits intovllm-project:mainfrom
ywang96:fix-budget-get
Feb 3, 2026
Merged

[Bugfix] Fix mm budget setting for Qwen Omni models#33634
DarkLight1337 merged 2 commits intovllm-project:mainfrom
ywang96:fix-budget-get

Conversation

@ywang96
Copy link
Copy Markdown
Member

@ywang96 ywang96 commented Feb 3, 2026

Purpose

If a model has hybrid-modality setting (e.g., use_audio_in_video=True for Qwen Omni model series) then multiple modalities can share the same placeholder in MultiModalInputs. This PR makes the modality checking more robust while add a comment to indicate so.

Fixes #33630

Test Plan

Test Result


Essential Elements of an Effective PR Description Checklist
  • The purpose of the PR, such as "Fix some issue (link existing issues this PR will resolve)".
  • The test plan, such as providing test command.
  • The test results, such as pasting the results comparison before and after, or e2e results
  • (Optional) The necessary documentation update, such as updating supported_models.md and examples for a new model.
  • (Optional) Release notes update. If your change is user facing, please update the release notes draft in the Google Doc.

Signed-off-by: Roger Wang <hey@rogerw.io>
@ywang96 ywang96 added the ready ONLY add when PR is ready to merge/full CI is needed label Feb 3, 2026
@mergify mergify bot added multi-modality Related to multi-modality (#4194) qwen Related to Qwen models bug Something isn't working labels Feb 3, 2026
Copy link
Copy Markdown
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

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

Code Review

This pull request correctly fixes a potential KeyError during multi-modal budget calculation for models that share placeholders between modalities, such as Qwen Omni. The change filters out active modalities without independent placeholder tokens, which is a simple and effective solution to prevent a crash. The implementation is sound and I have no further recommendations.

Copy link
Copy Markdown
Member

@DarkLight1337 DarkLight1337 left a comment

Choose a reason for hiding this comment

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

Sorry I broke this!

@DarkLight1337 DarkLight1337 enabled auto-merge (squash) February 3, 2026 03:39
@DarkLight1337 DarkLight1337 merged commit 10546f9 into vllm-project:main Feb 3, 2026
43 checks passed
PiratePai pushed a commit to PiratePai/epd_shm that referenced this pull request Feb 3, 2026
Signed-off-by: Roger Wang <hey@rogerw.io>
Signed-off-by: Pai <416932041@qq.com>
PiratePai pushed a commit to PiratePai/epd_shm that referenced this pull request Feb 3, 2026
Signed-off-by: Roger Wang <hey@rogerw.io>
Signed-off-by: Pai <416932041@qq.com>
linyueqian added a commit to linyueqian/vllm that referenced this pull request Feb 3, 2026
Signed-off-by: linyueqian <linyueqian@outlook.com>
gameofdimension pushed a commit to gameofdimension/vllm that referenced this pull request Feb 5, 2026
Signed-off-by: Roger Wang <hey@rogerw.io>
Signed-off-by: felix01.yu <felix01.yu@vipshop.com>
ItzDEXX pushed a commit to ItzDEXX/vllm that referenced this pull request Feb 19, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working multi-modality Related to multi-modality (#4194) qwen Related to Qwen models ready ONLY add when PR is ready to merge/full CI is needed

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[CI Failure]: Test Models (Qwen OMni)

2 participants