From ebc6b46c58ab13cbafb23c0926acc4e351f2d1d4 Mon Sep 17 00:00:00 2001
From: onlybetheone <iriszhangchong@gmail.com>
Date: 星期三, 15 二月 2023 20:37:57 +0800
Subject: [PATCH] add infer decoding model param
---
funasr/bin/asr_inference_mfcca.py | 3 +++
1 files changed, 3 insertions(+), 0 deletions(-)
diff --git a/funasr/bin/asr_inference_mfcca.py b/funasr/bin/asr_inference_mfcca.py
index 0da66c5..ac41c27 100644
--- a/funasr/bin/asr_inference_mfcca.py
+++ b/funasr/bin/asr_inference_mfcca.py
@@ -470,6 +470,7 @@
ngram_weight: float = 0.9,
nbest: int = 1,
num_workers: int = 1,
+ param_dict: dict = None,
**kwargs,
):
assert check_argument_types()
@@ -520,6 +521,8 @@
def _forward(data_path_and_name_and_type,
raw_inputs: Union[np.ndarray, torch.Tensor] = None,
output_dir_v2: Optional[str] = None,
+ fs: dict = None,
+ param_dict: dict = None,
):
# 3. Build data-iterator
if data_path_and_name_and_type is None and raw_inputs is not None:
--
Gitblit v1.9.1