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

Update xtts.py

This commit is contained in:
Moctar Haiz 2024-04-07 19:08:21 +02:00 committed by GitHub
parent 8a078be695
commit 4b0b906817
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
TTS/tts/models/xtts.py
View File

@ -410,15 +410,13 @@ class Xtts(BaseTTS):
if speaker_id is not None: if speaker_id is not None:
gpt_cond_latent, speaker_embedding = self.speaker_manager.speakers[speaker_id].values() gpt_cond_latent, speaker_embedding = self.speaker_manager.speakers[speaker_id].values()
return self.inference(text, language, gpt_cond_latent, speaker_embedding, **settings) return self.inference(text, language, gpt_cond_latent, speaker_embedding, **settings)
settings.update( settings.update({
{ "gpt_cond_len": config.gpt_cond_len,
"gpt_cond_len": config.gpt_cond_len, "gpt_cond_chunk_len": config.gpt_cond_chunk_len,
"gpt_cond_chunk_len": config.gpt_cond_chunk_len, "max_ref_len": config.max_ref_len,
"max_ref_len": config.max_ref_len, "sound_norm_refs": config.sound_norm_refs,
"sound_norm_refs": config.sound_norm_refs, })
} return self.full_inference(text, speaker_wav, language, **settings)
)
return self.full_inference(text, speaker_wav, language, speed, **settings)
@torch.inference_mode() @torch.inference_mode()
def full_inference( def full_inference(
@ -761,12 +759,13 @@ class Xtts(BaseTTS):
model_path = checkpoint_path or os.path.join(checkpoint_dir, "model.pth") model_path = checkpoint_path or os.path.join(checkpoint_dir, "model.pth")
vocab_path = vocab_path or os.path.join(checkpoint_dir, "vocab.json") vocab_path = vocab_path or os.path.join(checkpoint_dir, "vocab.json")
speaker_file_path = speaker_file_path or os.path.join(checkpoint_dir, "speakers_xtts.pth")
if speaker_file_path is None and checkpoint_dir is not None:
speaker_file_path = os.path.join(checkpoint_dir, "speakers_xtts.pth")
self.language_manager = LanguageManager(config) self.language_manager = LanguageManager(config)
self.speaker_manager = None self.speaker_manager = None
if os.path.exists(speaker_file_path): if speaker_file_path is not None and os.path.exists(speaker_file_path):
self.speaker_manager = SpeakerManager(speaker_file_path)
if os.path.exists(vocab_path): if os.path.exists(vocab_path):
self.tokenizer = VoiceBpeTokenizer(vocab_file=vocab_path) self.tokenizer = VoiceBpeTokenizer(vocab_file=vocab_path)