injectlib/index-tts
1
0
Fork 0
mirror of https://github.com/index-tts/index-tts.git synced 2025-11-25 19:37:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: Empty generator -> IndexError problem on non-streaming infer()

Browse Source
This commit is contained in:
Yt Zhong
2025-10-01 03:25:28 +08:00
parent b0c6ab8a93
commit 750d9d9d15
1 changed files with 12 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
indextts/infer_v2.py
View File

@@ -351,13 +351,16 @@ class IndexTTS2:
verbose, max_text_tokens_per_segment, stream_return, more_segment_before, **generation_kwargs
)
else:
return list(self.infer_generator(
spk_audio_prompt, text, output_path,
emo_audio_prompt, emo_alpha,
emo_vector,
use_emo_text, emo_text, use_random, interval_silence,
verbose, max_text_tokens_per_segment, stream_return, more_segment_before, **generation_kwargs
))[0]
try:
return list(self.infer_generator(
spk_audio_prompt, text, output_path,
emo_audio_prompt, emo_alpha,
emo_vector,
use_emo_text, emo_text, use_random, interval_silence,
verbose, max_text_tokens_per_segment, stream_return, more_segment_before, **generation_kwargs
))[0]
except IndexError:
return None
def infer_generator(self, spk_audio_prompt, text, output_path,
emo_audio_prompt=None, emo_alpha=1.0,
@@ -677,14 +680,14 @@ class IndexTTS2:
print(">> wav file saved to:", output_path)
if stream_return:
return None
return output_path
yield output_path
else:
if stream_return:
return None
# 返回以符合Gradio的格式要求
wav_data = wav.type(torch.int16)
wav_data = wav_data.numpy().T
return (sampling_rate, wav_data)
yield (sampling_rate, wav_data)
def find_most_similar_cosine(query_vector, matrix):