From dd17fb00af8f95bf454dc89ba56ad0e62cc3ca78 Mon Sep 17 00:00:00 2001 From: Jairo Correa Date: Wed, 3 May 2023 17:21:35 -0300 Subject: [PATCH] bug in env var USE_SMALL_MODELS (#236) --- bark/generation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bark/generation.py b/bark/generation.py index e4a6b2c..a704820 100644 --- a/bark/generation.py +++ b/bark/generation.py @@ -137,7 +137,7 @@ def _grab_best_device(use_gpu=True): def _get_ckpt_path(model_type, use_small=False): key = model_type - if use_small: + if use_small or USE_SMALL_MODELS: key += "_small" return os.path.join(CACHE_DIR, REMOTE_MODEL_PATHS[key]["file_name"])