diff --git a/funasr/tasks/asr.py b/funasr/tasks/asr.py index b01aa6ffa..387b25fa8 100644 --- a/funasr/tasks/asr.py +++ b/funasr/tasks/asr.py @@ -280,6 +280,8 @@ class ASRTask(AbsTask): postencoder_choices, # --decoder and --decoder_conf decoder_choices, + # --predictor and --predictor_conf + predictor_choices, ] # If you need to modify train() or eval() procedures, change Trainer class here