Skip to content

vllm.assets.audio

ASSET_DIR module-attribute

ASSET_DIR = 'multimodal_asset'

AudioAssetName module-attribute

AudioAssetName = Literal['winning_call', 'mary_had_lamb']

AudioAsset dataclass

Source code in vllm/assets/audio.py
@dataclass(frozen=True)
class AudioAsset:
    name: AudioAssetName

    @property
    def filename(self) -> str:
        return f"{self.name}.ogg"

    @property
    def audio_and_sample_rate(self) -> tuple[npt.NDArray, float]:
        audio_path = get_vllm_public_assets(filename=self.filename,
                                            s3_prefix=ASSET_DIR)
        return librosa.load(audio_path, sr=None)

    def get_local_path(self) -> Path:
        return get_vllm_public_assets(filename=self.filename,
                                      s3_prefix=ASSET_DIR)

    @property
    def url(self) -> str:
        return urljoin(VLLM_S3_BUCKET_URL, f"{ASSET_DIR}/{self.name}.ogg")

audio_and_sample_rate property

audio_and_sample_rate: tuple[NDArray, float]

filename property

filename: str

name instance-attribute

url property

url: str

__init__

__init__(name: AudioAssetName) -> None

get_local_path

get_local_path() -> Path
Source code in vllm/assets/audio.py
def get_local_path(self) -> Path:
    return get_vllm_public_assets(filename=self.filename,
                                  s3_prefix=ASSET_DIR)