Skip to content

[Feature]Migrate helios example - #4569

Merged
princepride merged 4 commits into
vllm-project:mainfrom
princepride:migrate-helios-example
Jun 21, 2026
Merged

princepride merged 4 commits into
vllm-project:mainfrom
princepride:migrate-helios-example

Conversation

@princepride

@princepride princepride commented Jun 19, 2026 •

Copy link
Copy Markdown
Collaborator

PLEASE FILL IN THE PR DESCRIPTION HERE.

Purpose

#4543

Test Plan

Helios-Base

python examples/offline_inference/text_to_video/text_to_video.py \
  --model BestWishYsh/Helios-Base \
  --prompt "A dynamic time-lapse of scenery rushing past the window of a speeding train."

Helios-Distilled

python examples/offline_inference/text_to_video/text_to_video.py \
  --model BestWishYsh/Helios-Distilled \
  --guidance-scale 1.0 \
  --extra-body '{"is_enable_stage2": true, "pyramid_num_inference_steps_list": [2, 2, 2], "is_amplify_first_chunk": true}'

Helios-Mid

python examples/offline_inference/text_to_video/text_to_video.py \
  --model BestWishYsh/Helios-Mid \
  --extra-body '{"is_enable_stage2": true, "pyramid_num_inference_steps_list": [20, 20, 20], "use_cfg_zero_star": true, "use_zero_init": true, "zero_steps": 1}'

Test Result

helios_output.mp4
helios_output.mp4
helios_output.mp4

princepride and others added 3 commits June 18, 2026 11:07
Register HeliosPipeline and HeliosPyramidPipeline in model_extras
(pyramid/staged sampling, CFG-zero* controls, and image/video
conditioning knobs), so the model-specific extra_args route through
extra_body and Helios can be driven via the standard task example.
Adds CPU unit test covering both class names.

Part of vllm-project#4539 (sub-task vllm-project#4543).

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Signed-off-by: princepride <wangzhipeng628@gmail.com>
Add a generic --extra-body JSON flag to the text_to_video example (knobs
declared in vllm_omni/model_extras/ merged into sampling extra_args), plus
a Helios preset and detect rule. Documents the three Helios T2V variants
(Base / Mid+CFG-Zero* / Distilled+DMD) as --extra-body recipes.

Additive only: the Wan/Hunyuan/LTX2 paths are untouched. Deletion of the
bespoke examples/{offline,online}_serving/helios dirs is deferred because
it is wired into docs (.nav.yml, generated example page, recipes/Helios).

Part of vllm-project#4539 (sub-task vllm-project#4543).

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Signed-off-by: princepride <wangzhipeng628@gmail.com>
…ix docs

Complete the Helios migration:
- Delete examples/offline_inference/helios and examples/online_serving/helios.
- Fold online serving into examples/online_serving/text_to_video: add
  run_curl_helios.sh (base/mid-stage2/distilled presets via extra_params),
  run_server_helios.sh, and a Helios README section.
- Remove the Helios nav entry from docs/.nav.yml and delete the generated
  docs/user_guide/examples/offline_inference/helios.md page.
- Repoint recipes/Helios/Helios.md at the shared text_to_video example using
  --extra-body for the Mid/Distilled knobs.

Part of vllm-project#4539 (sub-task vllm-project#4543).

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Signed-off-by: princepride <wangzhipeng628@gmail.com>
@chatgpt-codex-connector

Copy link
Copy Markdown

Codex usage limits have been reached for code reviews. Please check with the admins of this repo to increase the limits by adding credits.
Credits must be used to enable repository wide code reviews.

Signed-off-by: princepride <wangzhipeng628@gmail.com>

@hsliuustc0106 hsliuustc0106 left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

Example migration looks correct.

@lishunyang12 lishunyang12 added the ready label to trigger buildkite CI label Jun 21, 2026
@princepride
princepride enabled auto-merge (squash) June 21, 2026 07:05
@princepride
princepride merged commit 950e66a into vllm-project:main Jun 21, 2026
7 of 8 checks passed
@zzehli zzehli mentioned this pull request Jun 22, 2026
khairulkabir1661 pushed a commit to khairulkabir1661/vllm-omni that referenced this pull request Sep 25, 2026
Signed-off-by: princepride <wangzhipeng628@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ready label to trigger buildkite CI

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants