Mirrors/bark-with-voice-clone
1
0
Fork 0
mirror of https://github.com/serp-ai/bark-with-voice-clone.git synced 2025-12-16 11:48:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

add option for smaller models

Browse Source
This commit is contained in:
Georg Kucsko
2023-04-21 15:31:36 -04:00
parent 9751cfbfc4
commit c372430112
1 changed files with 33 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
bark/generation.py
View File

@@ -21,6 +21,7 @@ if (
torch.cuda.is_available() and torch.cuda.is_available() and
hasattr(torch.cuda, "amp") and hasattr(torch.cuda, "amp") and
hasattr(torch.cuda.amp, "autocast") and hasattr(torch.cuda.amp, "autocast") and
hasattr(torch.cuda, "is_bf16_supported") and
torch.cuda.is_bf16_supported() torch.cuda.is_bf16_supported()
): ):
autocast = funcy.partial(torch.cuda.amp.autocast, dtype=torch.bfloat16) autocast = funcy.partial(torch.cuda.amp.autocast, dtype=torch.bfloat16)
@@ -80,20 +81,36 @@ default_cache_dir = os.path.join(os.path.expanduser("~"), ".cache")
CACHE_DIR = os.path.join(os.getenv("XDG_CACHE_HOME", default_cache_dir), "suno", "bark_v0") CACHE_DIR = os.path.join(os.getenv("XDG_CACHE_HOME", default_cache_dir), "suno", "bark_v0")
USE_SMALL_MODELS = os.environ.get("SUNO_USE_SMALL_MODELS", False)
REMOTE_BASE_URL = "https://dl.suno-models.io/bark/models/v0/" REMOTE_BASE_URL = "https://dl.suno-models.io/bark/models/v0/"
if USE_SMALL_MODELS:
REMOTE_MODEL_PATHS = { REMOTE_MODEL_PATHS = {
"text": { "text": {
"path": os.environ.get("SUNO_TEXT_MODEL_PATH", os.path.join(REMOTE_BASE_URL, "text_2.pt")), "path": os.path.join(REMOTE_BASE_URL, "text.pt"),
"checksum": "b3e42bcbab23b688355cd44128c4cdd3",
},
"coarse": {
"path": os.path.join(REMOTE_BASE_URL, "coarse.pt"),
"checksum": "5fe964825e3b0321f9d5f3857b89194d",
},
"fine": {
"path": os.path.join(REMOTE_BASE_URL, "fine.pt"),
"checksum": "5428d1befe05be2ba32195496e58dc90",
},
}
else:
REMOTE_MODEL_PATHS = {
"text": {
"path": os.path.join(REMOTE_BASE_URL, "text_2.pt"),
"checksum": "54afa89d65e318d4f5f80e8e8799026a", "checksum": "54afa89d65e318d4f5f80e8e8799026a",
}, },
"coarse": { "coarse": {
"path": os.environ.get( "path": os.path.join(REMOTE_BASE_URL, "coarse_2.pt"),
"SUNO_COARSE_MODEL_PATH", os.path.join(REMOTE_BASE_URL, "coarse_2.pt")
),
"checksum": "8a98094e5e3a255a5c9c0ab7efe8fd28", "checksum": "8a98094e5e3a255a5c9c0ab7efe8fd28",
}, },
"fine": { "fine": {
"path": os.environ.get("SUNO_FINE_MODEL_PATH", os.path.join(REMOTE_BASE_URL, "fine_2.pt")), "path": os.path.join(REMOTE_BASE_URL, "fine_2.pt"),
"checksum": "59d184ed44e3650774a2f0503a48a97b", "checksum": "59d184ed44e3650774a2f0503a48a97b",
}, },
} }