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

add text ibest writer for SenseVoiceL

Browse Source
This commit is contained in:
志浩 2024-09-24 15:54:42 +08:00
parent d45230c6ba
commit 797bd57f91
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
funasr/models/sense_voice/model_small.py
View File

@ -1972,4 +1972,12 @@ class SenseVoiceL(nn.Module):
results.append(result_i) results.append(result_i)
ibest_writer = None
if kwargs.get("output_dir") is not None:
if not hasattr(self, "writer"):
self.writer = DatadirWriter(kwargs.get("output_dir"))
ibest_writer = self.writer[f"1best_recog"]
if ibest_writer is not None:
ibest_writer["text"][key[0]] = text
return results, meta_data return results, meta_data