From 57cb6329e25842b15c273c4f9179cb7ec86b9a35 Mon Sep 17 00:00:00 2001 From: JiehangXie Date: Mon, 6 Mar 2023 11:46:39 +0800 Subject: [PATCH] fix ort_predict_e2e.py --- paddlespeech/t2s/exps/ort_predict_e2e.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddlespeech/t2s/exps/ort_predict_e2e.py b/paddlespeech/t2s/exps/ort_predict_e2e.py index 3296d0e1e..5e4d273ed 100644 --- a/paddlespeech/t2s/exps/ort_predict_e2e.py +++ b/paddlespeech/t2s/exps/ort_predict_e2e.py @@ -77,7 +77,7 @@ def ort_predict(args): else: phone_ids = np.random.randint(1, 266, size=(T, )) am_input_feed.update({'text': phone_ids}) - if am_dataset in {"aishell3", "vctk", "mix"}: + if am_dataset in {"aishell3", "vctk", "mix", "canton"}: am_input_feed.update({'spk_id': spk_id}) elif am_name == 'speedyspeech': phone_ids = np.random.randint(1, 92, size=(T, )) @@ -112,7 +112,7 @@ def ort_predict(args): part_phone_ids = phone_ids[i].numpy() if am_name == 'fastspeech2': am_input_feed.update({'text': part_phone_ids}) - if am_dataset in {"aishell3", "vctk", "mix"}: + if am_dataset in {"aishell3", "vctk", "mix", "canton"}: am_input_feed.update({'spk_id': spk_id}) elif am_name == 'speedyspeech': part_tone_ids = frontend_dict['tone_ids'][i].numpy()