diff --git a/download-model.py b/download-model.py index 9332972..52c7a79 100644 --- a/download-model.py +++ b/download-model.py @@ -115,7 +115,7 @@ def get_download_links_from_huggingface(model, branch): 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(".*\.bin", fname) + is_ggml = re.match("ggml.*\.bin", fname) is_tokenizer = re.match("tokenizer.*\.model", fname) is_text = re.match(".*\.(txt|json|py|md)", fname) or is_tokenizer