Small fix

Signed-off-by: Xinyuan Li <xli257@b17.clsp.jhu.edu>
Trusted-AI · Jan 16, 2024 · 8f3205b · 8f3205b
1 parent a984b93
commit 8f3205b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/estimators/speech_recognition/test_pytorch_icefall.py b/tests/estimators/speech_recognition/test_pytorch_icefall.py
@@ -46,6 +46,7 @@ def test_pytorch_icefall(art_warning, expected_values, device_type):
 
         # load_model_ensemble
         transducer_model = get_transducer_model(params)
+        transducer_model.device = 'cpu'
         word2ids = get_word2id(params)
         get_id2word = get_id2word(params)
         model_ensemble = {