Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix issue #2971

This commit is contained in:
Edresson Casanova 2023-10-18 15:14:26 -03:00
parent 94dcf84979
commit 1f92741d6a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/tts/layers/xtts/gpt.py
View File

@ -559,7 +559,7 @@ class GPT(nn.Module):
bos_token_id=self.start_audio_token, bos_token_id=self.start_audio_token,
pad_token_id=self.stop_audio_token, pad_token_id=self.stop_audio_token,
eos_token_id=self.stop_audio_token, eos_token_id=self.stop_audio_token,
max_length=self.max_mel_tokens * 2 + self.max_prompt_tokens + self.max_text_tokens, max_length=self.max_mel_tokens,
**hf_generate_kwargs, **hf_generate_kwargs,
) )
if "return_dict_in_generate" in hf_generate_kwargs: if "return_dict_in_generate" in hf_generate_kwargs:
@ -572,7 +572,7 @@ class GPT(nn.Module):
bos_token_id=self.start_audio_token, bos_token_id=self.start_audio_token,
pad_token_id=self.stop_audio_token, pad_token_id=self.stop_audio_token,
eos_token_id=self.stop_audio_token, eos_token_id=self.stop_audio_token,
max_length=self.max_mel_tokens * 2 + self.max_prompt_tokens + self.max_text_tokens, max_length=self.max_mel_tokens,
do_stream=True, do_stream=True,
**hf_generate_kwargs, **hf_generate_kwargs,
) )