mirror of
https://github.com/zenlm/zen-router.git
synced 2026-10-30 14:57:44 +00:00
zeekay
f888212afb
Reads a prompt, picks the best model. 0.6B encoder-pooled router with task/route/feature heads; SFT from eval profiles + usage ledger, GRPO on quality - lambda*cost - mu*latency; serves via hanzo-engine and plugs the hanzo-router Classifier and enso Featurizer seams. Apache-2.0.
26 lines
766 B
Python
26 lines
766 B
Python
"""Upload trained zen-router artifacts to Hugging Face (zenlm/zen-router)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
from pathlib import Path
|
|
|
|
from huggingface_hub import HfApi
|
|
|
|
|
|
def main() -> None:
|
|
ap = argparse.ArgumentParser()
|
|
ap.add_argument("--repo", default="zenlm/zen-router")
|
|
ap.add_argument("--path", type=Path, default=Path("out/zen-router"))
|
|
args = ap.parse_args()
|
|
|
|
api = HfApi()
|
|
api.create_repo(args.repo, exist_ok=True, repo_type="model")
|
|
api.upload_folder(folder_path=str(args.path), repo_id=args.repo)
|
|
api.upload_file(path_or_fileobj="README.md", path_in_repo="README.md", repo_id=args.repo)
|
|
print(f"uploaded {args.path} -> https://huggingface.co/{args.repo}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|