From f6d34e1dd3903b6ea6072e391af941618169c2a4 Mon Sep 17 00:00:00 2001 From: mrq Date: Wed, 15 Mar 2023 01:09:20 +0000 Subject: [PATCH] and maybe I should have actually tested with ./models/tokenizers/ made --- src/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/utils.py b/src/utils.py index 08c537a..eeaa165 100755 --- a/src/utils.py +++ b/src/utils.py @@ -1734,7 +1734,7 @@ def get_diffusion_models(dir="./models/finetunes/", prefixed=False): return relative_paths([ get_model_path('diffusion_decoder.pth') ]) def get_tokenizer_jsons( dir="./models/tokenizers/" ): - additionals = sorted([ f'{additional_path}/{d}' for d in os.listdir(dir) if d[-5:] == ".json" ]) if os.path.isdir(dir) else [] + additionals = sorted([ f'{dir}/{d}' for d in os.listdir(dir) if d[-5:] == ".json" ]) if os.path.isdir(dir) else [] return relative_paths([ "./modules/tortoise-tts/tortoise/data/tokenizer.json" ] + additionals) def tokenize_text( text ):