[Misc] Optional installation of audio related packages (#8063)

This commit is contained in:
Roger Wang
2024-09-01 14:46:57 -07:00
committed by GitHub
parent 5231f0898e
commit 5b86b19954
6 changed files with 29 additions and 10 deletions

View File

@@ -8,7 +8,6 @@ from functools import lru_cache
from typing import (Iterable, List, Literal, Mapping, Optional, Tuple,
TypedDict, Union, cast)
import librosa
import numpy as np
import torch
import torch.utils.checkpoint
@@ -107,6 +106,11 @@ def input_mapper_for_ultravox(ctx: InputContext, data: object):
feature_extractor = whisper_feature_extractor(ctx)
if sr != feature_extractor.sampling_rate:
try:
import librosa
except ImportError:
raise ImportError(
"Please install vllm[audio] for audio support.") from None
audio = librosa.resample(audio,
orig_sr=sr,
target_sr=feature_extractor.sampling_rate)