From b6c4e35fd8e0350f57f022a2021d521cfbd4626a Mon Sep 17 00:00:00 2001 From: zappityzap Date: Sat, 2 Nov 2024 11:02:49 -0700 Subject: [PATCH] exclude tokenizer from list --- modules/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/utils.py b/modules/utils.py index 1971afa8..6e33b460 100644 --- a/modules/utils.py +++ b/modules/utils.py @@ -78,7 +78,7 @@ def get_available_models(): for file in files: file_path = os.path.join(dirpath, file) relative_path = os.path.relpath(file_path, shared.args.model_dir) - if not file.endswith(('.txt', '-np', '.pt', '.json', '.yaml', '.py')) and 'llama-tokenizer' not in file: + if not file.endswith(('.txt', '-np', '.pt', '.json', '.yaml', '.py')) and 'llama-tokenizer' not in relative_path: model_list.append(relative_path) return ['None'] + sorted(model_list, key=natural_keys)