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_demo.py 

Might be a good idea to also pass the checkpoint_dir as xtts.py seems to require speaker_file_path or checkpoint_dir and neither is passed in the demo
This commit is contained in:
Nick Potafiy 2024-01-30 01:15:06 -05:00 committed by GitHub
parent 5dcc16d193
commit baa7870fa1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
TTS/demos/xtts_ft_demo/xtts_demo.py
View File

@ -33,7 +33,8 @@ def load_model(xtts_checkpoint, xtts_config, xtts_vocab):
config.load_json(xtts_config)
XTTS_MODEL = Xtts.init_from_config(config)
print("Loading XTTS model! ")
XTTS_MODEL.load_checkpoint(config, checkpoint_path=xtts_checkpoint, vocab_path=xtts_vocab, use_deepspeed=False)
xtts_checkpoint_dir = os.path.dirname(xtts_checkpoint)
XTTS_MODEL.load_checkpoint(config, checkpoint_path=xtts_checkpoint, checkpoint_dir=xtts_checkpoint_dir, vocab_path=xtts_vocab, use_deepspeed=False)
if torch.cuda.is_available():
XTTS_MODEL.cuda()