Merge branch 'main' into MMdct-qk-norm

wenet-e2e · Aug 7, 2024 · 4a1ad11 · 4a1ad11
2 parents 3c2723a + 972b6fb
commit 4a1ad11
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git 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
@@ -3,8 +3,11 @@
 
 from wenet.text.hugging_face_tokenizer import HuggingFaceTokenizer
 
-os.system('pip install --no-input transformers==4.40.1')
-import transformers  # noqa
+try:
+    import transformers  # noqa
+except ImportError:
+    os.system('pip install --no-input transformers==4.40.1')
+    import transformers  # noqa
 
 
 @pytest.fixture(params=["bert-base-cased"])