From 19174842b82505c4627f095ea910ea0a9c998e1c Mon Sep 17 00:00:00 2001 From: Florian Kusche Date: Sun, 26 Mar 2023 19:41:14 +0200 Subject: [PATCH] Also download Markdown files --- download-model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/download-model.py b/download-model.py index 7ca33b7d..25386e5f 100644 --- a/download-model.py +++ b/download-model.py @@ -118,7 +118,7 @@ def get_download_links_from_huggingface(model, branch): is_safetensors = re.match("model.*\.safetensors", fname) is_pt = re.match(".*\.pt", fname) is_tokenizer = re.match("tokenizer.*\.model", fname) - is_text = re.match(".*\.(txt|json|py)", fname) or is_tokenizer + is_text = re.match(".*\.(txt|json|py|md)", fname) or is_tokenizer if any((is_pytorch, is_safetensors, is_pt, is_tokenizer, is_text)): if is_text: