浏览代码

Update ggml template

oobabooga 2 年之前
父节点
当前提交
875de5d983
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      download-model.py

+ 1 - 1
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