From af346ddd4d4486d8d1d1b1622be563a0f790487f Mon Sep 17 00:00:00 2001
From: root <root@localhost.localdomain>
Date: 星期六, 06 五月 2023 18:55:32 +0800
Subject: [PATCH] add multiple process test for python websocket client

---
 funasr/bin/build_trainer.py |    3 ++-
 1 files changed, 2 insertions(+), 1 deletions(-)

diff --git a/funasr/bin/build_trainer.py b/funasr/bin/build_trainer.py
index 94f7262..5c30fdb 100644
--- a/funasr/bin/build_trainer.py
+++ b/funasr/bin/build_trainer.py
@@ -83,7 +83,8 @@
         finetune_configs = yaml.safe_load(f)
         # set data_types
         if dataset_type == "large":
-            finetune_configs["dataset_conf"]["data_types"] = "sound,text"
+            if 'data_types' not in finetune_configs['dataset_conf']:
+                finetune_configs["dataset_conf"]["data_types"] = "sound,text"
     finetune_configs = update_dct(configs, finetune_configs)
     for key, value in finetune_configs.items():
         if hasattr(args, key):

--
Gitblit v1.9.1