Views
No views yet
1git clone https://github.com/FireRedTeam/FireRedTTS2.git
2cd FireRedTTS21conda create --name fireredtts2 python==3.11
2conda activate fireredtts2
3
4# Step 1. PyTorch Installation (if required)
5pip install torch==2.7.1 torchvision==0.22.1 torchaudio==2.7.1 --index-url https://download.pytorch.org/whl/cu126
6
7# Step 2. Install Dependencies
8pip install -e .
9pip install -r requirements.txt1git lfs install
2git clone https://huggingface.co/FireRedTeam/FireRedTTS2 pretrained_models/FireRedTTS2python gradio_demo.py --pretrained-dir "./pretrained_models/FireRedTTS2"1import os
2import sys
3import torch
4import torchaudio
5from fireredtts2.fireredtts2 import FireRedTTS2
6
7device = "cuda"
8
9fireredtts2 = FireRedTTS2(
10 pretrained_dir="./pretrained_models/FireRedTTS2",
11 gen_type="dialogue",
12 device=device,
13)
14
15text_list = [
16 "[S1]那可能说对对，没有去过美国来说去去看到美国线下。巴斯曼也好，沃尔玛也好，他们线下不管说，因为深圳出去的还是电子周边的会表达，会发现哇对这个价格真的是很高呀。都是卖三十五美金、四十美金，甚至一个手机壳，就是二十五美金开。",
17 "[S2]对，没错，我每次都觉得不不可思议。我什么人会买三五十美金的手机壳？但是其实在在那个target啊，就塔吉特这种超级市场，大家都是这样的，定价也很多人买。",
18 "[S1]对对，那这样我们再去看说亚马逊上面卖卖卖手机壳也好啊，贴膜也好，还包括说车窗也好，各种线材也好，大概就是七块九九或者说啊八块九九，这个价格才是卖的最多的啊。因为亚马逊的游戏规则限定的。如果说你卖七块九九以下，那你基本上是不赚钱的。",
19 "[S2]那比如说呃除了这个可能去到海外这个调查，然后这个调研考察那肯定是最直接的了。那平时我知道你是刚才建立了一个这个叫做呃rean的这样的一个一个播客，它是一个英文的。然后平时你还听一些什么样的东西，或者是从哪里获取一些这个海外市场的一些信息呢？",
20 "[S1]嗯，因为做做亚马逊的话呢，我们会关注很多行业内的东西。就比如说行业有什么样亚马逊有什么样新的游戏规则呀。呃，物流的价格有没有波动呀，包括说有没有什么新的评论的政策呀，广告有什么新的打法呀？那这些我们会会关关注很多行业内部的微信公众号呀，还包括去去查一些知乎专栏的文章呀，以及说我们周边有很多同行。那我们经常会坐在一起聊天，看看信息有什么共享。那这个是关注内内的一个方式。",
21]
22prompt_wav_list = [
23 "examples/chat_prompt/zh/S1.flac",
24 "examples/chat_prompt/zh/S2.flac",
25]
26
27prompt_text_list = [
28 "[S1]啊，可能说更适合美国市场应该是什么样子。那这这个可能说当然如果说有有机会能亲身的去考察去了解一下，那当然是有更好的帮助。",
29 "[S2]比如具体一点的，他觉得最大的一个跟他预想的不一样的是在什么地方。",
30]
31
32all_audio = fireredtts2.generate_dialogue(
33 text_list=text_list,
34 prompt_wav_list=prompt_wav_list,
35 prompt_text_list=prompt_text_list,
36 temperature=0.9,
37 topk=30,
38)
39torchaudio.save("chat_clone.wav", all_audio, 24000)1import os
2import sys
3import torch
4import torchaudio
5from fireredtts2.fireredtts2 import FireRedTTS2
6
7device = "cuda"
8lines = [
9 "Hello everyone, welcome to our newly launched FireRedTTS2. It supports multiple languages including English, Chinese, Japanese, Korean, French, German, and Russian. Additionally, this TTS model features long-context dialogue generation capabilities.",
10 "如果你厌倦了千篇一律的AI音色，不满意于其他模型语言支持不够丰富，那么本项目将会成为你绝佳的工具。",
11 "ランダムな話者と言語を選択して合成できます",
12 "이는 많은 인공지능 시스템에 유용합니다. 예를 들어, 제가 다양한 음성 데이터를 대량으로 생성해 여러분의 ASR 모델이나 대화 모델에 풍부한 데이터를 제공할 수 있습니다.",
13 "J'évolue constamment et j'espère pouvoir parler davantage de langues avec plus d'aisance à l'avenir.",
14]
15
16fireredtts2 = FireRedTTS2(
17 pretrained_dir="./pretrained_models/FireRedTTS2",
18 gen_type="monologue",
19 device=device,
20)
21
22# random speaker
23for i in range(len(lines)):
24 text = lines[i].strip()
25 audio = fireredtts2.generate_monologue(text=text)
26 # adjust temperature & topk
27 # audio = fireredtts2.generate_monologue(text=text, temperature=0.8, topk=30)
28 torchaudio.save(str(i) + ".wav", audio.cpu(), 24000)
29
30
31# # voice clone
32# for i in range(len(lines)):
33# text = lines[i].strip()
34
35# audio = fireredtts2.generate_monologue(
36# text=text,
37# prompt_wav=<prompt_wav_path>,
38# prompt_text=<prompt_wav_text>,
39# )
40# torchaudio.save(str(i) + ".wav", audio.cpu(), 24000)