pass ut on cpu

wenet-e2e · Feb 19, 2024 · 6f1fb43 · 6f1fb43
1 parent 0f967f8
commit 6f1fb43
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/wenet/ssl/w2vbert/test_w2vbert.py b/test/wenet/ssl/w2vbert/test_w2vbert.py
@@ -11,7 +11,7 @@
     from fairseq2.memory import MemoryBlock
 except ImportError:
     import os
-    os.system('pip install --no-input fairseq2')
+    os.system('pip install --no-input fairseq2 torchaudio==2.1.1')
     import fairseq2  # noqa
     from fairseq2.data.audio import AudioDecoder, WaveformToFbankConverter
     from fairseq2.memory import MemoryBlock