Quellcode durchsuchen

Update ggml template

oobabooga vor 2 Jahren
Ursprung
Commit
875de5d983
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  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