new model repo

RainPlayer · Apr 28, 2023 · 27ff4f9 · 27ff4f9
1 parent b24dd26
commit 27ff4f9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/bark/generation.py b/bark/generation.py
@@ -93,17 +93,17 @@ def autocast():
 
 REMOTE_MODEL_PATHS = {
     "text_small": {
-        "repo_id": "reach-vb/bark-small",
+        "repo_id": "reach-vb/bark",
         "file_name": "text.pt",
         "checksum": "b3e42bcbab23b688355cd44128c4cdd3",
     },
     "coarse_small": {
-        "repo_id": "reach-vb/bark-small",
+        "repo_id": "reach-vb/bark",
         "file_name": "coarse.pt",
         "checksum": "5fe964825e3b0321f9d5f3857b89194d",
     },
     "fine_small": {
-        "repo_id": "reach-vb/bark-small",
+        "repo_id": "reach-vb/bark",
         "file_name": "fine.pt",
         "checksum": "5428d1befe05be2ba32195496e58dc90",
     },