Files
vllm-omni/vllm_omni/diffusion/models/interface.py
T
2026-03-13 21:53:06 +08:00

24 lines
556 B
Python

# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
from typing import (
ClassVar,
Protocol,
runtime_checkable,
)
@runtime_checkable
class SupportImageInput(Protocol):
support_image_input: ClassVar[bool] = True
color_format: ClassVar[str] = "RGB" # Default color format
@runtime_checkable
class SupportAudioInput(Protocol):
support_audio_input: ClassVar[bool] = True
@runtime_checkable
class SupportAudioOutput(Protocol):
support_audio_output: ClassVar[bool] = True