"""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)