bug fix

shixian.shi

2023-03-13 0766a89c8d4a4d43238689fb4c4f72b241ebf437

bug fix

 funasr/bin/tp_inference.py

@@ -18,7 +18,7 @@

from funasr.fileio.datadir_writer import DatadirWriter
from funasr.datasets.preprocessor import LMPreprocessor
from funasr.tasks.asr import ASRTaskAligner as ASRTask
from funasr.tasks.asr import ASRTaskAligner_temp as ASRTask
from funasr.torch_utils.device_funcs import to_device
from funasr.torch_utils.set_all_random_seed import set_all_random_seed
from funasr.utils import config_argparse

 funasr/tasks/asr.py

@@ -1331,4 +1331,13 @@
            cls, train: bool = True, inference: bool = False
    ) -> Tuple[str, ...]:
        retval = ("speech", "text")
        return retval


class ASRTaskAligner(ASRTaskParaformer):
    @classmethod
    def required_data_names(
            cls, train: bool = True, inference: bool = False
    ) -> Tuple[str, ...]:
        retval = ("speech", "text")
        return retval