From 59df30cde17480ca6ae502331786605023446a45 Mon Sep 17 00:00:00 2001 From: Georg Kucsko Date: Wed, 3 May 2023 21:58:21 -0400 Subject: [PATCH] bool casting of env vars --- bark/generation.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/bark/generation.py b/bark/generation.py index a704820..42887e0 100644 --- a/bark/generation.py +++ b/bark/generation.py @@ -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 = {