Skip to content

vllm_omni.model_extras.registry ¶

ImageToImagePromptBuilder `module-attribute` ¶

ImageToImagePromptBuilder = Callable[
    [
        str,
        str | None,
        "Image.Image | list[Image.Image]",
        int | None,
        int | None,
    ],
    dict[str, Any],
]

TextToImagePromptBuilder `module-attribute` ¶

TextToImagePromptBuilder = Callable[
    [str, str | None, int | None, int | None],
    dict[str, Any],
]

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]

default_image_to_image_prompt ¶

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

default_text_to_image_prompt ¶

default_text_to_image_prompt(
    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