File size: 587 Bytes
b01bf09 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | """Download the files a models.json entry needs: python tools/fetch.py <id> [<id>...] (no ids = list)"""
import json, sys
from pathlib import Path
from huggingface_hub import hf_hub_download
ROOT = Path(__file__).resolve().parent.parent
reg = {m["id"]: m for m in json.load(open(ROOT / "models.json"))}
if len(sys.argv) < 2:
print("\n".join(reg)); sys.exit(0)
for i in sys.argv[1:]:
m = reg[i]; dest = ROOT / "models" / m["repo"].split("/")[1]
for f in m["files"]:
print("fetch", m["repo"], f, file=sys.stderr)
hf_hub_download(m["repo"], f, local_dir=dest)
|