fix CI

k2-fsa · Oct 29, 2024 · cf75bb4 · cf75bb4
1 parent 865ecc6
commit cf75bb4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/egs/gigaspeech/ASR/pruned_transducer_stateless2/decode.py b/egs/gigaspeech/ASR/pruned_transducer_stateless2/decode.py
@@ -260,7 +260,7 @@ def decode_one_batch(
       Return the decoding result. See above description for the format of
       the returned dict.
     """
-    device = model.device
+    device = next(model.parameters()).device
     feature = batch["inputs"]
     assert feature.ndim == 3