Merge pull request #1 from eurecom-asp/dev

Checkpoint inference and numpy versioning in setup tools

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

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

coqui_tts/pyproject.toml

@@ -1,5 +1,5 @@
 [build-system]
-requires = ["setuptools", "wheel", "cython==0.29.30", "numpy==1.22.0", "packaging"]
+requires = ["setuptools", "wheel", "cython==0.29.30", "numpy>=1.22.0", "packaging"]
 
 [flake8]
 max-line-length=120