| """Upload a prepared release folder atomically to main, preserving stable.""" |
| import argparse |
| from pathlib import Path |
| from huggingface_hub import HfApi |
| from model import load_model |
|
|
| def main(): |
| p=argparse.ArgumentParser(__doc__) |
| p.add_argument('--folder',default='release') |
| p.add_argument('--repo',default='User-2468/mini-unet-colorizer') |
| p.add_argument('--expected-main',default=None,help='Optional optimistic-concurrency commit SHA') |
| a=p.parse_args() |
| folder=Path(a.folder) |
| model=load_model(folder) |
| assert sum(p.numel() for p in model.parameters())<4_000_000 |
| api=HfApi() |
| refs={x.name:x.target_commit for x in api.list_repo_refs(a.repo).branches} |
| stable=refs.get('stable') |
| result=api.upload_folder(repo_id=a.repo,revision='main',folder_path=folder, |
| parent_commit=a.expected_main or refs['main'], |
| commit_message='Update validated colorizer weights, spatial decoder and deployment artifacts', |
| ignore_patterns=['__pycache__/**','*.pyc']) |
| after={x.name:x.target_commit for x in api.list_repo_refs(a.repo).branches} |
| if stable!=after.get('stable'):raise RuntimeError('Stable changed concurrently; inspect repository history') |
| print(result.commit_url) |
|
|
| if __name__=='__main__':main() |
|
|
