diff --git a/egs/librispeech/ASR/.run_v3.sh.swp b/egs/librispeech/ASR/.run_v3.sh.swp index 25547da9d..a86e47ad1 100644 Binary files a/egs/librispeech/ASR/.run_v3.sh.swp and b/egs/librispeech/ASR/.run_v3.sh.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_encoder.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_encoder.py.swp index 24c01e477..2bd280635 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_encoder.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_encoder.py.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_encoder.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_encoder.py index a4e33cba0..7a933dd1f 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_encoder.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_encoder.py @@ -20,6 +20,15 @@ from encoder_interface import EncoderInterface from torch import Tensor, nn from icefall.utils import make_pad_mask, subsequent_chunk_mask +try: + import fairseq + from data2vec_audio import * +except Exception as e: + print("Error: FairSeq is not properly installed.") + print( + "Please install FairSeq: cd ${MAIN_ROOT}/tools && make fairseq.done" + ) + raise e class FairSeqData2VecEncoder(EncoderInterface): @@ -47,17 +56,7 @@ class FairSeqData2VecEncoder(EncoderInterface): assert check_argument_types() super().__init__() - if w2v_url != "": - try: - import fairseq - from data2vec_audio import * - except Exception as e: - print("Error: FairSeq is not properly installed.") - print( - "Please install FairSeq: cd ${MAIN_ROOT}/tools && make fairseq.done" - ) - raise e - + ''' if os.path.exists('/home/work/workspace/models/data2vec_model/audio_base_ls.pt'): self.w2v_model_path = '/home/work/workspace/models/data2vec_model/audio_base_ls.pt'