From b5ad7c81be2e24f255cac3d0ef0037bf88228366 Mon Sep 17 00:00:00 2001
From: Kun Zou <54402842+kenzoukun@users.noreply.github.com>
Date: 星期六, 21 十二月 2024 17:13:46 +0800
Subject: [PATCH] Support eparaformer model on aishell1 recipe (#2327)
---
funasr/datasets/sense_voice_datasets/datasets.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/datasets/sense_voice_datasets/datasets.py b/funasr/datasets/sense_voice_datasets/datasets.py
index 20232cc..2e0266e 100644
--- a/funasr/datasets/sense_voice_datasets/datasets.py
+++ b/funasr/datasets/sense_voice_datasets/datasets.py
@@ -325,8 +325,8 @@
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|>")
--
Gitblit v1.9.1