Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-29 10:59:32 +01:00
Code Issues Packages Projects Releases Wiki Activity

exclude tokenizer from list

Browse Source
This commit is contained in:
zappityzap 2024-11-02 11:02:49 -07:00
parent eaf9980c9a
commit b6c4e35fd8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/utils.py
View File

@ -78,7 +78,7 @@ def get_available_models():
for file in files: for file in files:
file_path = os.path.join(dirpath, file) file_path = os.path.join(dirpath, file)
relative_path = os.path.relpath(file_path, shared.args.model_dir) 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) model_list.append(relative_path)
return ['None'] + sorted(model_list, key=natural_keys) return ['None'] + sorted(model_list, key=natural_keys)