| #!/usr/bin/env bash |
| |
| set -euo pipefail |
|
|
| : "${HF_TOKEN:?Defina HF_TOKEN antes de executar}" |
| : "${HF_REPO_ID:?Defina HF_REPO_ID (ex: LesterCerioli/Speed)}" |
|
|
| MODEL_DIR=${MODEL_DIR:-output} |
| PRIVATE=${PRIVATE:-false} |
| COMMIT_MSG=${COMMIT_MSG:-"Upload LLM Fiscal Brasileiro checkpoint"} |
|
|
| echo "==> Publicando LLM Fiscal no Hugging Face Hub" |
| echo " repo : $HF_REPO_ID" |
| echo " modelo : $MODEL_DIR" |
| echo " privado: $PRIVATE" |
| echo "" |
|
|
| python3 - <<EOF |
| import os |
| from huggingface_hub import HfApi |
| |
| api = HfApi(token=os.environ["HF_TOKEN"]) |
| |
| api.create_repo( |
| repo_id=os.environ["HF_REPO_ID"], |
| repo_type="model", |
| private=os.environ.get("PRIVATE", "false") == "true", |
| exist_ok=True, |
| ) |
| |
| api.upload_folder( |
| folder_path=".", |
| repo_id=os.environ["HF_REPO_ID"], |
| repo_type="model", |
| ignore_patterns=[ |
| "*.pyc", |
| "**/__pycache__/**", |
| ".git/**", |
| "output/**", |
| "scripts/**", |
| ], |
| commit_message=os.environ.get("COMMIT_MSG", "Upload checkpoint"), |
| ) |
| |
| print(f"✅ Modelo publicado em: https://huggingface.co/{os.environ['HF_REPO_ID']}") |
| EOF |
|
|