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 default emo and event target for sensevoice

Browse Source
This commit is contained in:
北念 2024-07-18 15:05:28 +08:00
parent a98550fdf5
commit bd352983c6
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
funasr/datasets/sense_voice_datasets/datasets.py
View File

@ -325,8 +325,8 @@ class SenseVoiceCTCDataset(torch.utils.data.Dataset):
asr_target = item["target"]
if self.preprocessor_text:
asr_target = self.preprocessor_text(asr_target)
emo_target = item["emo_target"]
event_target = item["event_target"]
emo_target = item.get("emo_target", "<|NEUTRAL|>")
event_target = item.get("event_target", "<|Speech|>")
text_language = item.get("text_language", "<|zh|>")
punc_itn_bottom = item.get("with_or_wo_itn", "<|woitn|>")