shixian.shi %!s(int64=3) %!d(string=hai) anos
pai
achega
0766a89c8d
Modificáronse 2 ficheiros con 10 adicións e 1 borrados
  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