File size: 1,765 Bytes
c7deb87 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 | """Upload the sharded checkpoint folder to zhifeixie/Mini-Omni3."""
import argparse
from pathlib import Path
from huggingface_hub import HfApi, login
def main():
ap = argparse.ArgumentParser()
ap.add_argument("--folder", default="/Users/yansc-xzf/Desktop/工作/Mini-Omni3/github/omni3/Mini-Omni3/checkpoints")
ap.add_argument("--repo-id", default="zhifeixie/Mini-Omni3")
ap.add_argument("--path-in-repo", default=".",
help="Subdirectory inside the repo. Use '.' to upload at repo root.")
ap.add_argument("--private", action="store_false")
ap.add_argument("--token", default=None,
help="HF token. If omitted, uses cached login (`huggingface-cli login`).")
args = ap.parse_args()
if args.token:
login(token=args.token)
api = HfApi()
api.create_repo(repo_id=args.repo_id, repo_type="model",
private=args.private, exist_ok=True)
folder = Path(args.folder)
if not folder.is_dir():
raise SystemExit(f"Not a directory: {folder}")
print(f"Uploading {folder} → {args.repo_id}:{args.path_in_repo}")
# upload_large_folder is designed for multi-GB folders: parallel chunked
# uploads with automatic resume on interruption.
api.upload_large_folder(
repo_id=args.repo_id,
repo_type="model",
folder_path=str(folder),
# path_in_repo is not directly supported by upload_large_folder;
# we emulate it via allow_patterns if needed. The simplest path is
# to upload the folder as-is at the repo root, OR to rename your
# local folder to match the desired in-repo name.
)
print("Done. Check: https://huggingface.co/" + args.repo_id)
if __name__ == "__main__":
main() |