From 0766a89c8d4a4d43238689fb4c4f72b241ebf437 Mon Sep 17 00:00:00 2001
From: shixian.shi <shixian.shi@alibaba-inc.com>
Date: 星期一, 13 三月 2023 16:15:31 +0800
Subject: [PATCH] bug fix
---
funasr/tasks/asr.py | 9 +++++++++
funasr/bin/tp_inference.py | 2 +-
2 files changed, 10 insertions(+), 1 deletions(-)
diff --git a/funasr/bin/tp_inference.py b/funasr/bin/tp_inference.py
index e374a22..8c9c0f3 100644
--- a/funasr/bin/tp_inference.py
+++ b/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
diff --git a/funasr/tasks/asr.py b/funasr/tasks/asr.py
index 13898a6..05eace7 100644
--- a/funasr/tasks/asr.py
+++ b/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
\ No newline at end of file
--
Gitblit v1.9.1