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(xtts): load tokenizer file based on config as last resort

This commit is contained in:
Enno Hermann 2024-07-05 13:52:01 +02:00
parent c1a929b720
commit 9192ef1aa6
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,6 +1,7 @@
import logging import logging
import os import os
from dataclasses import dataclass from dataclasses import dataclass
from pathlib import Path
import librosa import librosa
import torch import torch
@ -760,7 +761,11 @@ 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") if vocab_path is None:
if checkpoint_dir is not None and (Path(checkpoint_dir) / "vocab.json").is_file():
vocab_path = str(Path(checkpoint_dir) / "vocab.json")
else:
vocab_path = config.model_args.tokenizer_file
if speaker_file_path is None and checkpoint_dir is not None: if speaker_file_path is None and checkpoint_dir is not None:
speaker_file_path = os.path.join(checkpoint_dir, "speakers_xtts.pth") speaker_file_path = os.path.join(checkpoint_dir, "speakers_xtts.pth")