suno-ai
/
bark
mirror of https://github.com/suno-ai/bark.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

bool casting of env vars

pull/247/head
Georg Kucsko 2023-05-03 21:58:21 -04:00
parent dd17fb00af
commit 59df30cde1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
bark/generation.py
View File

@ -85,9 +85,9 @@ 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")
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)
USE_SMALL_MODELS = bool(os.environ.get("SUNO_USE_SMALL_MODELS", False))
GLOBAL_ENABLE_MPS = bool(os.environ.get("SUNO_ENABLE_MPS", False))
OFFLOAD_CPU = bool(os.environ.get("SUNO_OFFLOAD_CPU", False))
REMOTE_MODEL_PATHS = {