funasr/tasks/asr.py
@@ -103,7 +103,7 @@ name="specaug", classes=dict( specaug=SpecAug, specaug_lfr=SpecAugLFR, specaug_lfr=FSpecAugLR, ), type_check=AbsSpecAug, default=None, @@ -1684,7 +1684,7 @@ # 7. Build model if encoder.unified_model_training: if hasattr(encoder, 'unified_model_training') and encoder.unified_model_training: model = UnifiedTransducerModel( vocab_size=vocab_size, token_list=token_list,