File size: 1,459 Bytes
e568bec |
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 |
import os
import sys
from huggingface_hub import HfApi, create_repo, upload_folder
from config import Config
def main():
print("🚀 开始上传所有 Checkpoint 到 robot4/emotion ...")
api = HfApi()
try:
user_info = api.whoami()
username = user_info['name']
print(f"✅ User: {username}")
except:
print("❌ Please login first.")
return
# 1. 目标仓库
repo_id = f"{username}/emotion"
print(f"📦 目标仓库: {repo_id}")
create_repo(repo_id=repo_id, repo_type="model", exist_ok=True)
# 2. 上传整个 results 目录
# 我们会上传 results/checkpoint-500, results/checkpoint-1000, etc.
# 也就是在仓库根目录下会有这些文件夹
results_dir = Config.RESULTS_DIR
print(f"⬆️ 正在上传 {results_dir} 下的所有模型文件...")
print(" (已自动忽略 optimizer.pt 等大文件以节省时间和流量)")
upload_folder(
folder_path=results_dir,
repo_id=repo_id,
repo_type="model",
# 排除非必要大文件
ignore_patterns=["optimizer.pt", "scheduler.pt", "rng_state.pth", "*.zip"]
)
print(f"🎉 所有模型上传完成！查看地址: https://huggingface.co/{repo_id}")
if __name__ == "__main__":
current_dir = os.path.dirname(os.path.abspath(__file__))
parent_dir = os.path.dirname(current_dir)
sys.path.append(parent_dir)
main()
|