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 extract token run_mode

Browse Source
This commit is contained in:
志浩 2024-09-24 17:06:52 +08:00
parent 0fa8e2976f
commit 1fb762d9be
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -2052,6 +2052,7 @@ class SenseVoiceL(nn.Module):
ark_writer = None
if kwargs.get("output_dir") is not None:
out_dir = kwargs.get("output_dir")
os.makedirs(out_dir, exist_ok=True)
if not hasattr(self, "writer"):
out_path = os.path.join(out_dir, f"enc_token")
self.writer = kaldiio.WriteHelper(f"ark,scp,f:{out_path}.ark,{out_path}.scp")