from huggingface_hub import login
login()  # 会提示你输入token

from huggingface_hub import HfApi, create_repo
import os

# 设置你的信息
username = "你的用户名"
model_name = "reward-model-qwen"
repo_id = f"{username}/{model_name}"

# 创建API实例
api = HfApi()

# 创建仓库
try:
    create_repo(repo_id, repo_type="model", private=False)
    print(f"创建仓库: {repo_id}")
except Exception as e:
    print(f"仓库已存在或创建失败: {e}")

# 上传模型文件
model_path = "./reward_model_output/final_model"
if os.path.exists(model_path):
    api.upload_folder(
        folder_path=model_path,
        repo_id=repo_id,
        repo_type="model",
        commit_message="上传训练好的奖励模型 v1.0"
    )
    print(f"模型已上传到: https://huggingface.co/{repo_id}")
else:
    print(f"模型路径不存在: {model_path}")