From abb33d6b2097e5b0643326bc1b376a63cdc2f967 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 24 六月 2024 17:06:21 +0800
Subject: [PATCH] Dev gzf deepspeed (#1844)
---
funasr/datasets/audio_datasets/index_ds.py | 6 ++++++
1 files changed, 6 insertions(+), 0 deletions(-)
diff --git a/funasr/datasets/audio_datasets/index_ds.py b/funasr/datasets/audio_datasets/index_ds.py
index 385218a..39ef409 100644
--- a/funasr/datasets/audio_datasets/index_ds.py
+++ b/funasr/datasets/audio_datasets/index_ds.py
@@ -118,6 +118,12 @@
text_language = data.get("text_language", None)
if text_language is not None:
contents_i["text_language"] = text_language
+ if "emo_target" in data:
+ contents_i["emo_target"] = data["emo_target"]
+ if "event_target" in data:
+ contents_i["event_target"] = data["event_target"]
+ if "with_or_wo_itn" in data:
+ contents_i["with_or_wo_itn"] = data["with_or_wo_itn"]
# audio_language = data.get("audio_language", None)
# if audio_language is not None:
# contents_i["audio_language"] = audio_language
--
Gitblit v1.9.1