Skip to content

vllm_omni.diffusion.model_metadata

HUNYUAN_IMAGE3_MAX_INPUT_IMAGES module-attribute

HUNYUAN_IMAGE3_MAX_INPUT_IMAGES = 3

QWEN_IMAGE_EDIT_PLUS_MAX_INPUT_IMAGES module-attribute

QWEN_IMAGE_EDIT_PLUS_MAX_INPUT_IMAGES = 4

DiffusionModelMetadata dataclass

max_multimodal_image_inputs class-attribute instance-attribute

max_multimodal_image_inputs: int | None = None

supports_multimodal_inputs class-attribute instance-attribute

supports_multimodal_inputs: bool = False

get_diffusion_model_metadata

get_diffusion_model_metadata(
    model_class_name: str | None,
) -> DiffusionModelMetadata