From f555278709f48e4a690edfc9501d3564ec3e3cac Mon Sep 17 00:00:00 2001 From: Kappa Date: Wed, 22 Mar 2023 00:14:55 +0600 Subject: [PATCH] Update download-model.py --- lib/download-model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/download-model.py b/lib/download-model.py index 4001eff..0957820 100644 --- 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