From 6939705d1810fd9a4a0493e49676060dfa42ff0b Mon Sep 17 00:00:00 2001 From: mrfakename Date: Sat, 15 Jul 2023 13:17:25 -0700 Subject: [PATCH] Enable MPS (if available) by default on Mac --- bark/generation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bark/generation.py b/bark/generation.py index 54f9870..d4ec04d 100644 --- a/bark/generation.py +++ b/bark/generation.py @@ -90,7 +90,7 @@ def _cast_bool_env_var(s): USE_SMALL_MODELS = _cast_bool_env_var(os.environ.get("SUNO_USE_SMALL_MODELS", "False")) -GLOBAL_ENABLE_MPS = _cast_bool_env_var(os.environ.get("SUNO_ENABLE_MPS", "False")) +GLOBAL_ENABLE_MPS = _cast_bool_env_var(os.environ.get("SUNO_ENABLE_MPS", "True")) OFFLOAD_CPU = _cast_bool_env_var(os.environ.get("SUNO_OFFLOAD_CPU", "False"))