From 7f995679aa44137f5e4bd6b66a1b0519a9c7d3ee Mon Sep 17 00:00:00 2001 From: Georg Kucsko Date: Mon, 10 Apr 2023 13:00:33 -0400 Subject: [PATCH] update models --- bark/generation.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/bark/generation.py b/bark/generation.py index 4d4f604..91b322a 100644 --- a/bark/generation.py +++ b/bark/generation.py @@ -69,9 +69,11 @@ os.makedirs(CACHE_DIR, exist_ok=True) REMOTE_BASE_URL = "http://s3.amazonaws.com/suno-public/bark/models/v0/" REMOTE_MODEL_PATHS = { - "text": os.environ.get("SUNO_TEXT_MODEL_PATH", os.path.join(REMOTE_BASE_URL, "text.pt")), - "coarse": os.environ.get("SUNO_COARSE_MODEL_PATH", os.path.join(REMOTE_BASE_URL, "coarse.pt")), - "fine": os.environ.get("SUNO_FINE_MODEL_PATH", os.path.join(REMOTE_BASE_URL, "fine.pt")), + "text": os.environ.get("SUNO_TEXT_MODEL_PATH", os.path.join(REMOTE_BASE_URL, "text_2.pt")), + "coarse": os.environ.get( + "SUNO_COARSE_MODEL_PATH", os.path.join(REMOTE_BASE_URL, "coarse_2.pt") + ), + "fine": os.environ.get("SUNO_FINE_MODEL_PATH", os.path.join(REMOTE_BASE_URL, "fine_2.pt")), }