shixian.shi 3 years ago
parent
commit
0766a89c8d
2 changed files with 10 additions and 1 deletions
  1. 1 1
      funasr/bin/tp_inference.py
  2. 9 0
      funasr/tasks/asr.py

+ 1 - 1
funasr/bin/tp_inference.py

@@ -18,7 +18,7 @@ from typeguard import check_argument_types
 
 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

+ 9 - 0
funasr/tasks/asr.py

@@ -1326,6 +1326,15 @@ class ASRTaskAligner(ASRTaskParaformer):
         assert check_return_type(model)
         return model
 
+    @classmethod
+    def required_data_names(
+            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