35b6514ee5
- Introduced `de_tokenized_by_CJK_char` for restoring original text from tokenized format. - Added `TextTokenizer` class for improved tokenization, including sentence splitting and handling of special tokens. - Enhanced `TextNormalizer` to handle names and pinyin tones with placeholder mechanisms. - Added regression tests for new features in `regression_test.py`.
43 lines
3.6 KiB
Python
43 lines
3.6 KiB
Python
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 = "“我爱你！”的英语是“I love you!”"
|
||
tts.infer(audio_prompt=prompt_wav, text=text, output_path=f"outputs/{text[:20]}.wav", verbose=True)
|
||
text = "Joseph Gordon-Levitt is an American actor"
|
||
tts.infer(audio_prompt=prompt_wav, text=text, output_path=f"outputs/{text[:20]}.wav", verbose=True)
|
||
text = "约瑟夫·高登-莱维特是美国演员"
|
||
tts.infer(audio_prompt=prompt_wav, text=text, output_path=f"outputs/{text[:20]}.wav", verbose=True)
|
||
text = "蒂莫西·唐纳德·库克（英文名：Timothy Donald Cook），通称蒂姆·库克（Tim Cook），现任苹果公司首席执行官。"
|
||
tts.infer(audio_prompt=prompt_wav, text=text, output_path="outputs/蒂莫西·唐纳德·库克.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)
|
||
# 长文本推理测试
|
||
text = """《盗梦空间》是由美国华纳兄弟影片公司出品的电影，由克里斯托弗·诺兰执导并编剧，
|
||
莱昂纳多·迪卡普里奥、玛丽昂·歌迪亚、约瑟夫·高登-莱维特、艾利奥特·佩吉、汤姆·哈迪等联袂主演，
|
||
2010年7月16日在美国上映，2010年9月1日在中国内地上映，2020年8月28日在中国内地重映。
|
||
影片剧情游走于梦境与现实之间，被定义为“发生在意识结构内的当代动作科幻片”，
|
||
讲述了由莱昂纳多·迪卡普里奥扮演的造梦师，带领特工团队进入他人梦境，从他人的潜意识中盗取机密，并重塑他人梦境的故事。
|
||
""".replace("\n", "")
|
||
tts.infer_fast(audio_prompt=prompt_wav, text=text, output_path=f"outputs/{text[:20]}.wav", verbose=True)
|