Spaces:
Running on Zero
Running on Zero
File size: 3,028 Bytes
9e874de | 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 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 | """Upload a trained Objectverse Diary LoRA adapter folder to Hugging Face Hub."""
from __future__ import annotations
import argparse
import json
from pathlib import Path
from typing import Any
REQUIRED_ADAPTER_FILES = ("adapter_config.json",)
ADAPTER_WEIGHT_FILES = ("adapter_model.safetensors", "adapter_model.bin")
def validate_adapter_dir(adapter_dir: Path) -> dict[str, object]:
if not adapter_dir.exists() or not adapter_dir.is_dir():
raise FileNotFoundError(f"Adapter directory does not exist: {adapter_dir}")
missing = [name for name in REQUIRED_ADAPTER_FILES if not (adapter_dir / name).exists()]
has_weights = any((adapter_dir / name).exists() for name in ADAPTER_WEIGHT_FILES)
if not has_weights:
missing.append("adapter_model.safetensors or adapter_model.bin")
if missing:
raise ValueError(f"Adapter directory is missing required files: {', '.join(missing)}")
files = sorted(path.name for path in adapter_dir.iterdir() if path.is_file())
return {
"adapter_dir": str(adapter_dir),
"files": files,
"file_count": len(files),
}
def upload_adapter(
*,
adapter_dir: Path,
repo_id: str,
private: bool,
commit_message: str,
dry_run: bool,
) -> dict[str, object]:
summary = validate_adapter_dir(adapter_dir)
summary.update(
{
"repo_id": repo_id,
"private": private,
"commit_message": commit_message,
"dry_run": dry_run,
}
)
if dry_run:
summary["uploaded"] = False
return summary
from huggingface_hub import HfApi
api = HfApi()
api.create_repo(repo_id=repo_id, repo_type="model", private=private, exist_ok=True)
api.upload_folder(
folder_path=str(adapter_dir),
repo_id=repo_id,
repo_type="model",
commit_message=commit_message,
)
summary["uploaded"] = True
summary["url"] = f"https://huggingface.co/{repo_id}"
return summary
def _print_json(payload: dict[str, Any]) -> None:
print(json.dumps(payload, indent=2, sort_keys=True), flush=True)
def _parse_args() -> argparse.Namespace:
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument("--adapter-dir", type=Path, required=True)
parser.add_argument("--repo-id", required=True)
parser.add_argument("--private", action="store_true")
parser.add_argument(
"--commit-message",
default="Upload Objectverse Diary LoRA adapter",
)
parser.add_argument("--dry-run", action="store_true")
return parser.parse_args()
def main() -> None:
args = _parse_args()
_print_json(
upload_adapter(
adapter_dir=args.adapter_dir,
repo_id=args.repo_id,
private=args.private,
commit_message=args.commit_message,
dry_run=args.dry_run,
)
)
if __name__ == "__main__":
try:
main()
except Exception as exc:
raise SystemExit(str(exc)) from exc
|