From ac3a7568a7e34e2f90e98b9ee6a31425fc9fe66f Mon Sep 17 00:00:00 2001 From: vaibhavs10 Date: Thu, 27 Apr 2023 16:19:58 +0200 Subject: [PATCH] up --- bark/generation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/bark/generation.py b/bark/generation.py index 3b12757..a3c8a0b 100644 --- a/bark/generation.py +++ b/bark/generation.py @@ -90,7 +90,7 @@ USE_SMALL_MODELS = os.environ.get("SUNO_USE_SMALL_MODELS", False) GLOBAL_ENABLE_MPS = os.environ.get("SUNO_ENABLE_MPS", False) OFFLOAD_CPU = os.environ.get("SUNO_OFFLOAD_CPU", False) -# REMOTE_BASE_URL = "https://dl.suno-models.io/bark/models/v0/" +REMOTE_BASE_URL = "https://dl.suno-models.io/bark/models/v0/" # REMOTE_MODEL_PATHS = { # "text_small": { @@ -176,7 +176,7 @@ def _md5(fname): def _get_ckpt_path(model_type, use_small=False): model_key = f"{model_type}_small" if use_small or USE_SMALL_MODELS else model_type - model_name = _string_md5(REMOTE_MODEL_PATHS[model_key]["path"]) + model_name = _string_md5(REMOTE_MODEL_PATHS[model_key]["file_name"]) return os.path.join(CACHE_DIR, f"{model_name}.pt")