From ca0886395b90aef52dadac8cf0002d0ba297fa76 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Mon, 26 Sep 2022 23:58:02 +0200 Subject: [PATCH] Add warning for torchaudio <= 0.10 in MCTCTFeatureExtractor (#19203) Co-authored-by: ydshieh --- src/transformers/models/mctct/feature_extraction_mctct.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/transformers/models/mctct/feature_extraction_mctct.py b/src/transformers/models/mctct/feature_extraction_mctct.py index 573551bcf7..9785d530e9 100644 --- a/src/transformers/models/mctct/feature_extraction_mctct.py +++ b/src/transformers/models/mctct/feature_extraction_mctct.py @@ -21,6 +21,7 @@ from typing import List, Optional, Union import numpy as np import torch import torchaudio +from packaging import version from ...feature_extraction_sequence_utils import SequenceFeatureExtractor from ...feature_extraction_utils import BatchFeature @@ -30,6 +31,13 @@ from ...utils import logging logger = logging.get_logger(__name__) +parsed_torchaudio_version_base = version.parse(version.parse(torchaudio.__version__).base_version) +if not parsed_torchaudio_version_base >= version.parse("0.10"): + logger.warning( + f"You are using torchaudio=={torchaudio.__version__}, but torchaudio>=0.10.0 is required to use " + "MCTCTFeatureExtractor. This requires torch>=1.10.0. Please upgrade torch and torchaudio." + ) + class MCTCTFeatureExtractor(SequenceFeatureExtractor): r"""