Skip to content

vllm_omni.diffusion.model_metadata

BOOGU_IMAGE_MAX_INPUT_IMAGES module-attribute

BOOGU_IMAGE_MAX_INPUT_IMAGES = 1

FLUX2_KLEIN_MAX_INPUT_IMAGES module-attribute

FLUX2_KLEIN_MAX_INPUT_IMAGES = 4

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

attention_mask_free class-attribute instance-attribute

attention_mask_free: bool = False

final_output_type class-attribute instance-attribute

final_output_type: str | None = None

max_multimodal_image_inputs class-attribute instance-attribute

max_multimodal_image_inputs: int | None = None

supported_control_upload_types class-attribute instance-attribute

supported_control_upload_types: tuple[str, ...] = ()

supports_latent_mask_editing class-attribute instance-attribute

supports_latent_mask_editing: bool = False

supports_mixed_reference_inputs class-attribute instance-attribute

supports_mixed_reference_inputs: bool = False

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