diff --git a/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_H.py b/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_H.py index 14c20c5b2..5783be0d1 100755 --- a/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_H.py +++ b/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_H.py @@ -41,7 +41,7 @@ import kaldifeat import kaldifst import torch import torchaudio -from kaldi_hmm_gmm import DecodableCtc, FasterDecoder, FasterDecoderOptions +from kaldi_decoder import DecodableCtc, FasterDecoder, FasterDecoderOptions from torch.nn.utils.rnn import pad_sequence diff --git a/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_HL.py b/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_HL.py index f02655468..da99a3465 100755 --- a/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_HL.py +++ b/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_HL.py @@ -41,7 +41,7 @@ import kaldifeat import kaldifst import torch import torchaudio -from kaldi_hmm_gmm import DecodableCtc, FasterDecoder, FasterDecoderOptions +from kaldi_decoder import DecodableCtc, FasterDecoder, FasterDecoderOptions from torch.nn.utils.rnn import pad_sequence diff --git a/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_HLG.py b/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_HLG.py index d1e1c4af2..315db2541 100755 --- a/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_HLG.py +++ b/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_HLG.py @@ -40,7 +40,7 @@ import kaldifeat import kaldifst import torch import torchaudio -from kaldi_hmm_gmm import DecodableCtc, FasterDecoder, FasterDecoderOptions +from kaldi_decoder import DecodableCtc, FasterDecoder, FasterDecoderOptions from torch.nn.utils.rnn import pad_sequence diff --git a/requirements.txt b/requirements.txt index c031d683c..5a8326619 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ kaldifst kaldilm kaldialign -kaldi-hmm-gmm +kaldi-decoder sentencepiece>=0.1.96 tensorboard typeguard