diff --git a/indextts/infer.py b/indextts/infer.py index 0485061..6c7f5f0 100644 --- a/indextts/infer.py +++ b/indextts/infer.py @@ -184,7 +184,7 @@ class IndexTTS: return [ sentence.strip() for sentence in sentences if sentence.strip() and sentence.strip() not in {"'", ".", ","} ] - + def bucket_sentences(self, sentences, enable): """ Sentence data bucketing @@ -288,7 +288,6 @@ class IndexTTS: gpt_forward_time = 0 bigvgan_time = 0 - # text processing all_text_tokens = [] self._set_gr_progress(0.1, "text processing...") diff --git a/tests/regression_test.py b/tests/regression_test.py new file mode 100644 index 0000000..58d5be0 --- /dev/null +++ b/tests/regression_test.py @@ -0,0 +1,28 @@ +from indextts.infer import IndexTTS + +if __name__ == "__main__": + prompt_wav="tests/sample_prompt.wav" + tts = IndexTTS(cfg_path="checkpoints/config.yaml", model_dir="checkpoints", is_fp16=True, use_cuda_kernel=False) + # 单音频推理测试 + text="晕 XUAN4 是 一 种 GAN3 觉" + tts.infer(audio_prompt=prompt_wav, text=text, output_path=f"outputs/{text[:20]}.wav", verbose=True) + text='大家好,我现在正在bilibili 体验 ai 科技,说实话,来之前我绝对想不到!AI技术已经发展到这样匪夷所思的地步了!' + tts.infer(audio_prompt=prompt_wav, text=text, output_path=f"outputs/{text[:20]}.wav", verbose=True) + text="There is a vehicle arriving in dock number 7?" + tts.infer(audio_prompt=prompt_wav, text=text, output_path=f"outputs/{text[:20]}.wav", verbose=True) + + # 并行推理测试 + text="亲爱的伙伴们,大家好!每一次的努力都是为了更好的未来,要善于从失败中汲取经验,让我们一起勇敢前行,迈向更加美好的明天!" + tts.infer_fast(audio_prompt=prompt_wav, text=text, output_path=f"outputs/{text[:20]}.wav", verbose=True) + text="The weather is really nice today, perfect for studying at home.Thank you!" + tts.infer_fast(audio_prompt=prompt_wav, text=text, output_path=f"outputs/{text[:20]}.wav", verbose=True) + text='''叶远随口答应一声,一定帮忙云云。 +教授看叶远的样子也知道,这事情多半是黄了。 +谁得到这样的东西也不会轻易贡献出来,这是很大的一笔财富。 +叶远回来后,又自己做了几次试验,发现空间湖水对一些外伤也有很大的帮助。 +找来一只断了腿的兔子,喝下空间湖水,一天时间,兔子就完全好了。 +还想多做几次试验,可是身边没有试验的对象,就先放到一边,了解空间湖水可以饮用,而且对人有利,这些就足够了。 +感谢您的收听,下期再见! + '''.replace("\n", "") + tts.infer_fast(audio_prompt=prompt_wav, text=text, output_path=f"outputs/{text[:20]}.wav", verbose=True) + diff --git a/tests/sample_prompt.wav b/tests/sample_prompt.wav new file mode 100644 index 0000000..ccc65c2 Binary files /dev/null and b/tests/sample_prompt.wav differ