This commit is contained in:
shixian.shi 2023-06-27 17:24:19 +08:00
parent 1d1a33d5be
commit 92e14eb734

View File

@ -85,7 +85,9 @@ def build_trainer(modelscope_dict,
finetune_configs = yaml.safe_load(f)
# set data_types
if dataset_type == "large":
finetune_configs["dataset_conf"]["data_types"] = "sound,text"
# 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):