vllm_omni.model_extras ¶

Modules:

Name	Description
`bagel`
`registry`
`sensenova_u1`

build_image_to_image_prompt ¶

build_image_to_image_prompt(
    model_class_name: str | None,
    prompt: str,
    negative_prompt: str | None,
    input_image: Image | list[Image],
    height: int | None = None,
    width: int | None = None,
) -> dict[str, Any]

build_text_to_image_prompt ¶

build_text_to_image_prompt(
    model_class_name: str | None,
    prompt: str,
    negative_prompt: str | None,
    height: int | None = None,
    width: int | None = None,
) -> dict[str, Any]

get_extra_body_params ¶

get_extra_body_params(
    model_class_name: str | None,
) -> frozenset[str]

get_extra_output_params ¶

get_extra_output_params(
    model_class_name: str | None,
) -> frozenset[str]

get_model_class_name ¶

get_model_class_name(omni: Any) -> str | None

Extract model_class_name from an Omni/AsyncOmni instance.

This hides the internal ODConfig plumbing from example scripts.

should_init_extra_args_for_non_diffusion_stages ¶

should_init_extra_args_for_non_diffusion_stages(
    model_class_name: str | None,
) -> bool