From ead7bde1ab2ba939f0c3a73b3c829860d82888c8 Mon Sep 17 00:00:00 2001 From: Cyrus Leung Date: Wed, 4 Mar 2026 22:47:32 +0800 Subject: [PATCH] [Bugfix] Make `kaldi_native_fbank` optional (#35996) Signed-off-by: DarkLight1337 --- requirements/common.txt | 1 - requirements/test.in | 1 + requirements/test.txt | 2 ++ .../processors/fireredasr2_processor.py | 9 ++++++++- 4 files changed, 11 insertions(+), 2 deletions(-) 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__)