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

update

Browse Source
This commit is contained in:
嘉渊 2023-07-03 21:51:17 +08:00
parent 15a440f5da
commit ba3d455b21
2 changed files with 4 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
funasr/bin/asr_inference_launch.py
View File

@ -1349,10 +1349,7 @@ def inference_transducer(
left_context=left_context,
right_context=right_context,
)
speech2text = Speech2TextTransducer.from_pretrained(
model_tag=model_tag,
**speech2text_kwargs,
)
speech2text = Speech2TextTransducer(**speech2text_kwargs)
def _forward(data_path_and_name_and_type,
raw_inputs: Union[np.ndarray, torch.Tensor] = None,

15
funasr/bin/diar_inference_launch.py
View File

@ -92,10 +92,7 @@ def inference_sond(
embedding_node="resnet1_dense"
)
logging.info("speech2xvector_kwargs: {}".format(speech2xvector_kwargs))
speech2xvector = Speech2Xvector.from_pretrained(
model_tag=model_tag,
**speech2xvector_kwargs,
)
speech2xvector = Speech2Xvector(**speech2xvector_kwargs)
speech2xvector.sv_model.eval()
# 2b. Build speech2diar
@ -109,10 +106,7 @@ def inference_sond(
dur_threshold=dur_threshold,
)
logging.info("speech2diarization_kwargs: {}".format(speech2diar_kwargs))
speech2diar = Speech2DiarizationSOND.from_pretrained(
model_tag=model_tag,
**speech2diar_kwargs,
)
speech2diar = Speech2DiarizationSOND(**speech2diar_kwargs)
speech2diar.diar_model.eval()
def output_results_str(results: dict, uttid: str):
@ -257,10 +251,7 @@ def inference_eend(
dtype=dtype,
)
logging.info("speech2diarization_kwargs: {}".format(speech2diar_kwargs))
speech2diar = Speech2DiarizationEEND.from_pretrained(
model_tag=model_tag,
**speech2diar_kwargs,
)
speech2diar = Speech2DiarizationEEND(**speech2diar_kwargs)
speech2diar.diar_model.eval()
def output_results_str(results: dict, uttid: str):