diff --git a/anonymizer.py b/anonymizer.py index be34be47d2..eccd5e7f95 100644 --- a/anonymizer.py +++ b/anonymizer.py @@ -15,6 +15,10 @@ class Anonymizer(torch.nn.Module): def __init__(self, checkpoint_dir: str, voice_dirs: Union[list[str], None] = None): super().__init__() + + if not os.path.exists(checkpoint_dir): + print(f"Checkpoint directory {checkpoint_dir} not found, creating it") + os.makedirs(checkpoint_dir) # 1. initialize Bark config = BarkConfig() # don't change the custom config for the love of god diff --git a/coqui_tts/TTS/tts/configs/bark_config.py b/coqui_tts/TTS/tts/configs/bark_config.py index 4d1cd1374a..68d454cfbc 100644 --- a/coqui_tts/TTS/tts/configs/bark_config.py +++ b/coqui_tts/TTS/tts/configs/bark_config.py @@ -68,7 +68,8 @@ class BarkConfig(BaseTTSConfig): COARSE_SEMANTIC_PAD_TOKEN: int = 12_048 COARSE_INFER_TOKEN: int = 12_050 - REMOTE_BASE_URL = "https://huggingface.co/erogol/bark/tree/main/" + # REMOTE_BASE_URL = "https://huggingface.co/erogol/bark/tree/main/" # this used to return the actual model, now returns an HTML page lol + REMOTE_BASE_URL = "https://huggingface.co/erogol/bark/resolve/main/" REMOTE_MODEL_PATHS: Dict = None LOCAL_MODEL_PATHS: Dict = None SMALL_REMOTE_MODEL_PATHS: Dict = None