Replace ggml occurences with gguf

2023-08-26 01:06:59 -07:00 · 2023-08-26 01:06:59 -07:00 · 83640d6f43
commit 83640d6f43
parent 1a642c12b5
5 changed files with 12 additions and 12 deletions
--- a/modules/llamacpp_hf.py
+++ b/modules/llamacpp_hf.py
@ -165,7 +165,7 @@ class LlamacppHF(PreTrainedModel):
        if path.is_file():
            model_file = path
        else:
-            model_file = list(path.glob('*ggml*.bin'))[0]
+            model_file = list(path.glob('*.gguf*'))[0]

        logger.info(f"llama.cpp weights detected: {model_file}\n")