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
coqui-ai · Jan 30, 2024 · baa7870 · baa7870
1 parent 5dcc16d
commit baa7870
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/TTS/demos/xtts_ft_demo/xtts_demo.py b/TTS/demos/xtts_ft_demo/xtts_demo.py
@@ -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()