diff --git a/requirements.txt b/requirements.txt index e78b45af78..54d3e853bf 100644 --- a/requirements.txt +++ b/requirements.txt @@ -20,6 +20,6 @@ torchaudio>=2.1.2 tqdm deepspeed>=0.14.0 librosa -openai-whisper +openai-whisper==20231117 pre-commit==3.5.0 langid diff --git a/test/wenet/text/test_hugging_face_tokenizer.py b/test/wenet/text/test_hugging_face_tokenizer.py index 24caadbb8d..25d38e04a3 100644 --- a/test/wenet/text/test_hugging_face_tokenizer.py +++ b/test/wenet/text/test_hugging_face_tokenizer.py @@ -6,7 +6,7 @@ try: import transformers # noqa except ImportError: - os.system('pip install --no-input transformers') + os.system('pip install --no-input transformers==4.40.1') import transformers # noqa