We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent db94d4f commit c41bf84Copy full SHA for c41bf84
modules/models.py
@@ -258,7 +258,7 @@ def llamacpp_loader(model_name):
258
if path.is_file():
259
model_file = path
260
else:
261
- model_file = list(Path(f'{shared.args.model_dir}/{model_name}').glob('*ggml*.bin'))[0]
+ model_file = list(Path(f'{shared.args.model_dir}/{model_name}').glob('*ggml*'))[0]
262
263
logger.info(f"llama.cpp weights detected: {model_file}\n")
264
model, tokenizer = LlamaCppModel.from_pretrained(model_file)
0 commit comments