leon-ai/leon
1
1
Fork 0
mirror of https://github.com/leon-ai/leon.git synced 2024-11-30 19:07:39 +03:00
Code Issues Projects Releases Wiki Activity

fix(python tcp server): TTS inference correct params mapping

Browse Source
This commit is contained in:
louistiti 2024-06-21 09:51:52 +08:00
parent 8e28466cae
commit 3734a16b6a
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tcp_server/src/lib/tts/api.py
View File

@ -107,7 +107,7 @@ class TTS(nn.Module):
print(" > ===========================")
return texts
def tts_iter(self, text, speaker_id, sdp_ratio=0.2, noise_scale=0.6, noise_scale_w=0.8, speed=1.0, pbar=None, format=None, position=None, quiet=False, stream=False):
def tts_iter(self, text, speaker_id, sdp_ratio=0.2, noise_scale=0.6, noise_scale_w=0.8, speed=1.0, pbar=None, position=None, quiet=False, stream=False):
tic = time.perf_counter()
self.log(f"Generating audio for:\n{text}")
language = self.language
@ -169,7 +169,18 @@ class TTS(nn.Module):
def tts_to_file(self, text, speaker_id, output_path=None, sdp_ratio=0.2, noise_scale=0.6, noise_scale_w=0.8, speed=1.0, pbar=None, format=None, position=None, quiet=False, stream=False):
audio_list = []
for audio in self.tts_iter(text, speaker_id, sdp_ratio, noise_scale, noise_scale_w, speed, pbar, position, quiet, stream):
for audio in self.tts_iter(
text=text,
speaker_id=speaker_id,
sdp_ratio=sdp_ratio,
noise_scale=noise_scale,
noise_scale_w=noise_scale_w,
speed=speed,
pbar=pbar,
position=position,
quiet=quiet,
stream=stream
):
audio_list.append(audio)
audio = np.concatenate(audio_list)