diff --git a/modules/utils.py b/modules/utils.py index 6e33b460..c4a30950 100644 --- a/modules/utils.py +++ b/modules/utils.py @@ -73,12 +73,13 @@ def natural_keys(text): def get_available_models(): + exclude_extensions = ('.txt', '-np', '.pt', '.json', '.yaml', '.py', '.modelfile') model_list = [] for dirpath, dirnames, files in os.walk(shared.args.model_dir, followlinks=True): 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 relative_path: + if not file.endswith(exclude_extensions) and 'llama-tokenizer' not in relative_path: model_list.append(relative_path) return ['None'] + sorted(model_list, key=natural_keys)