Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-12-26 14:20:40 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix Qwen1.5 in llamacpp_HF

Browse Source
This commit is contained in:
oobabooga 2024-02-15 19:03:47 -08:00
parent 080f7132c0
commit b2b74c83a6
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/models.py
View File

@ -257,7 +257,7 @@ def llamacpp_HF_loader(model_name):
path = Path(f'{shared.args.model_dir}/{model_name}')
# Check if a HF tokenizer is available for the model
if all((path / file).exists() for file in ['tokenizer.model', 'tokenizer_config.json']):
if all((path / file).exists() for file in ['tokenizer.json', 'tokenizer_config.json']):
logger.info(f'Using tokenizer from: \"{path}\"')
else:
logger.error("Could not load the model because a tokenizer in Transformers format was not found.")