Sfoglia il codice sorgente

Update ggml template

oobabooga 2 anni fa
parent
commit
875de5d983
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  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