Skip to content

vllm_omni.model_extras.bagel ¶

BAGEL_EXTRA_BODY_PARAMS `module-attribute` ¶

BAGEL_EXTRA_BODY_PARAMS = frozenset(
    {
        "cfg_text_scale",
        "cfg_img_scale",
        "cfg_interval",
        "cfg_renorm_type",
        "cfg_renorm_min",
        "negative_prompt",
        "think",
        "max_think_tokens",
        "do_sample",
        "text_temperature",
        "timestep_shift",
    }
)

BAGEL_EXTRA_OUTPUT_PARAMS `module-attribute` ¶

BAGEL_EXTRA_OUTPUT_PARAMS = frozenset(
    {"text_output", "think_text"}
)

BAGEL_INIT_EXTRA_ARGS_FOR_NON_DIFFUSION_STAGES `module-attribute` ¶

BAGEL_INIT_EXTRA_ARGS_FOR_NON_DIFFUSION_STAGES = True

build_image_to_image_prompt ¶

build_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]

build_text_to_image_prompt ¶

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

build_x_to_text_prompt ¶

build_x_to_text_prompt(
    model: str, prompt: str, has_image: bool
) -> tuple[dict[str, Any], list[int] | None]

Build BAGEL's chat prompt for text-output tasks.