Update download-model.py

sashaok123 · Mar 21, 2023 · f555278 · f555278
1 parent cfa2d90
commit f555278
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/download-model.py b/lib/download-model.py
@@ -58,7 +58,7 @@ def get_download_links(url):
     download_links = []
     for link in soup.find_all("a"):
         href = link.get("href")
-        if href and any(href.endswith(ext) for ext in (".ckpt", ".safetensors", ".yaml", ".bin", ".pt")):
+        if href and any(href.endswith(ext) for ext in (".ckpt", ".safetensors", ".yaml", ".bin", ".pt", ".pth", ".json")):
             download_links.append("https://huggingface.co" + href.replace("/blob", "/resolve"))
 
     return download_links