vllm_omni.model_extras.helios ¶

HELIOS_EXTRA_BODY_PARAMS = frozenset(
    {
        "is_enable_stage2",
        "pyramid_num_stages",
        "pyramid_num_inference_steps_list",
        "is_amplify_first_chunk",
        "is_skip_first_chunk",
        "use_cfg_zero_star",
        "use_zero_init",
        "zero_steps",
        "image",
        "video",
        "add_noise_to_image_latents",
        "image_noise_sigma_min",
        "image_noise_sigma_max",
        "add_noise_to_video_latents",
        "video_noise_sigma_min",
        "video_noise_sigma_max",
    }
)

HELIOS_EXTRA_OUTPUT_PARAMS = frozenset()