diff --git a/requirements/common.txt b/requirements/common.txt index 9ee1b7151..ec7ce5df9 100644 --- a/requirements/common.txt +++ b/requirements/common.txt @@ -57,4 +57,3 @@ opentelemetry-sdk >= 1.27.0 opentelemetry-api >= 1.27.0 opentelemetry-exporter-otlp >= 1.27.0 opentelemetry-semantic-conventions-ai >= 0.4.1 -kaldi-native-fbank >= 1.18.7 diff --git a/requirements/test.in b/requirements/test.in index ed9bb4711..a551a4c05 100644 --- a/requirements/test.in +++ b/requirements/test.in @@ -67,6 +67,7 @@ segmentation-models-pytorch > 0.4.0 # Required for Prithvi tests gpt-oss >= 0.0.7; python_version > '3.11' perceptron # required for isaac test +kaldi-native-fbank >= 1.18.7 # required for fireredasr2 test # Newer versions of datasets require torchcoded, that makes the tests fail in CI because of a missing library. # Older versions are in conflict with teerratorch requirements. diff --git a/requirements/test.txt b/requirements/test.txt index 8aa2d6768..aacb8fbff 100644 --- a/requirements/test.txt +++ b/requirements/test.txt @@ -417,6 +417,8 @@ jsonschema-specifications==2024.10.1 # via jsonschema junit-xml==1.9 # via schemathesis +kaldi-native-fbank==1.22.3 + # via -r requirements/test.in kaleido==0.2.1 # via genai-perf kiwisolver==1.4.7 diff --git a/vllm/transformers_utils/processors/fireredasr2_processor.py b/vllm/transformers_utils/processors/fireredasr2_processor.py index 67c74ab15..98c99ec39 100644 --- a/vllm/transformers_utils/processors/fireredasr2_processor.py +++ b/vllm/transformers_utils/processors/fireredasr2_processor.py @@ -1,8 +1,8 @@ # SPDX-License-Identifier: Apache-2.0 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project +from typing import TYPE_CHECKING -import kaldi_native_fbank as knf import numpy as np import torch import torch.nn.functional as F @@ -16,6 +16,13 @@ from transformers.processing_utils import ProcessorMixin from transformers.utils import TensorType from vllm.logger import init_logger +from vllm.utils.import_utils import LazyLoader + +if TYPE_CHECKING: + import kaldi_native_fbank as knf +else: + knf = LazyLoader("knf", globals(), "kaldi_native_fbank") + logger = init_logger(__name__)