From 1436c5845a3a7d60106415fda4511ccbd54d77cf Mon Sep 17 00:00:00 2001 From: Wojtek Kowaluk Date: Thu, 4 May 2023 16:48:36 +0200 Subject: [PATCH] fix ggml detection regex in model downloader (#1779) --- download-model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/download-model.py b/download-model.py index a095f398..67696f3b 100644 --- a/download-model.py +++ b/download-model.py @@ -114,7 +114,7 @@ def get_download_links_from_huggingface(model, branch, text_only=False): is_pytorch = re.match("(pytorch|adapter)_model.*\.bin", fname) is_safetensors = re.match(".*\.safetensors", fname) is_pt = re.match(".*\.pt", fname) - is_ggml = re.match("ggml.*\.bin", fname) + is_ggml = re.match(".*ggml.*\.bin", fname) is_tokenizer = re.match("(tokenizer|ice).*\.model", fname) is_text = re.match(".*\.(txt|json|py|md)", fname) or is_tokenizer