speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

bug fix

Browse Source
This commit is contained in:
shixian.shi 2023-03-13 16:15:31 +08:00
parent 76fd90d230
commit 0766a89c8d
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/bin/tp_inference.py
View File

@ -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
funasr/tasks/asr.py
View File

@ -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