Skip to content

Commit c41bf84

Browse files
authored
Update models.py
1 parent db94d4f commit c41bf84

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modules/models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -258,7 +258,7 @@ def llamacpp_loader(model_name):
258258
if path.is_file():
259259
model_file = path
260260
else:
261-
model_file = list(Path(f'{shared.args.model_dir}/{model_name}').glob('*ggml*.bin'))[0]
261+
model_file = list(Path(f'{shared.args.model_dir}/{model_name}').glob('*ggml*'))[0]
262262

263263
logger.info(f"llama.cpp weights detected: {model_file}\n")
264264
model, tokenizer = LlamaCppModel.from_pretrained(model_file)

0 commit comments

Comments
 (0)