Skip to content

[BugFix] Bagel img2img e2e: drop extra_body height/width #3054

Merged
Gaohan123 merged 3 commits into
vllm-project:mainfrom
NumberWan:bugfix/bagle_size
Apr 23, 2026
Merged

[BugFix] Bagel img2img e2e: drop extra_body height/width #3054
Gaohan123 merged 3 commits into
vllm-project:mainfrom
NumberWan:bugfix/bagle_size

Conversation

@NumberWan
Copy link
Copy Markdown
Contributor

PLEASE FILL IN THE PR DESCRIPTION HERE ENSURING ALL CHECKLIST ITEMS (AT THE BOTTOM) HAVE BEEN CONSIDERED.

Purpose

Fixes CI failure reported in #3048.
After #2932, tests/e2e/online_serving/test_bagel_online.py::test_bagel_img2img_online started sending explicit height / width in extra_body. The shared diffusion response assertions then require the returned image pixels to exactly match those values.
For Bagel multistage img2img, the diffusion stage can follow KV metadata image_shape derived from the thinker/img2img preprocessing path, so the decoded output may not equal the extra_body width/height (e.g. CI observed 1024 width vs expected 512).
This PR removes height / width from the img2img e2e extra_body so the test continues to validate the online img2img path without asserting a strict output resolution contract that the stack does not currently guarantee.


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. Please provide the test scripts & test commands. Please state the reasons if your codes don't require additional test scripts. For test file guidelines, please check the test style doc
  • The test results. Please paste the results comparison before and after, or the e2e results.
  • (Optional) The necessary documentation update, such as updating supported_models.md and examples for a new model. Please run mkdocs serve to sync the documentation editions to ./docs.
  • (Optional) Release notes update. If your change is user-facing, please update the release notes draft.

BEFORE SUBMITTING, PLEASE READ https://github.com/vllm-project/vllm-omni/blob/main/CONTRIBUTING.md (anything written below this line will be removed by GitHub Actions)

Signed-off-by: NumberWan <wantszkin2003@gmail.com>
Copy link
Copy Markdown

@chatgpt-codex-connector chatgpt-codex-connector Bot left a comment

Choose a reason for hiding this comment

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

💡 Codex Review

Here are some automated review suggestions for this pull request.

Reviewed commit: 170d9c28ce

ℹ️ About Codex in GitHub

Codex has been enabled to automatically review pull requests in this repo. Reviews are triggered when you

  • Open a pull request for review
  • Mark a draft as ready
  • Comment "@codex review".

If Codex has suggestions, it will comment; otherwise it will react with 👍.

When you sign up for Codex through ChatGPT, Codex can also answer questions or update the PR, like "@codex address that feedback".

Comment thread tests/e2e/online_serving/test_bagel_online.py
@yenuo26 yenuo26 added merge-test label to trigger buildkite merge test CI ready label to trigger buildkite CI labels Apr 23, 2026
@Gaohan123 Gaohan123 added this to the v0.20.0 milestone Apr 23, 2026
Copy link
Copy Markdown
Collaborator

@Gaohan123 Gaohan123 left a comment

Choose a reason for hiding this comment

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

LGTM

@Gaohan123 Gaohan123 merged commit 1a7fb55 into vllm-project:main Apr 23, 2026
8 checks passed
lengrongfu pushed a commit to lengrongfu/vllm-omni that referenced this pull request May 1, 2026
clodaghwalsh17 pushed a commit to clodaghwalsh17/nm-vllm-omni-ent that referenced this pull request May 12, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

merge-test label to trigger buildkite merge test CI ready label to trigger buildkite CI

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants