Bug fix on XTTS inference

This commit is contained in:
Edresson Casanova 2023-11-24 12:07:00 -03:00
parent 3fc2880127
commit af74cd4426
1 changed files with 5 additions and 0 deletions

View File

@ -272,6 +272,11 @@ class Xtts(BaseTTS):
style_embs = [] style_embs = []
for i in range(0, audio.shape[1], 22050 * chunk_length): for i in range(0, audio.shape[1], 22050 * chunk_length):
audio_chunk = audio[:, i : i + 22050 * chunk_length] audio_chunk = audio[:, i : i + 22050 * chunk_length]
# if the chunk is too short ignore it
if audio_chunk.size(-1) < 22050 * 0.33:
continue
mel_chunk = wav_to_mel_cloning( mel_chunk = wav_to_mel_cloning(
audio_chunk, audio_chunk,
mel_norms=self.mel_stats.cpu(), mel_norms=self.mel_stats.cpu(),