From e5480abb7b28d5f4b3253153e295937d3e1c89a7 Mon Sep 17 00:00:00 2001 From: XTer Date: Sun, 10 Mar 2024 21:38:44 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=AD=A3=E4=BA=86=E9=94=99=E8=AF=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- GPT_SoVITS/TTS_infer_pack/TTS.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/GPT_SoVITS/TTS_infer_pack/TTS.py b/GPT_SoVITS/TTS_infer_pack/TTS.py index 9cc54911..dd25b1a6 100644 --- a/GPT_SoVITS/TTS_infer_pack/TTS.py +++ b/GPT_SoVITS/TTS_infer_pack/TTS.py @@ -650,12 +650,13 @@ class TTS: else: audio = [item for batch in audio for item in batch] - if type(audio) == list: - audio = np.array(audio) - audio = (audio * 32768).astype(np.int16) - else: + + try: audio = np.concatenate(audio, 0) audio = (audio * 32768).astype(np.int16) + except: + audio = np.array(audio) + audio = (audio * 32768).astype(np.int16) try: if speed_factor != 1.0: