import argparse
from huggingface_hub import HfApi

parser = argparse.ArgumentParser(description="Upload files to Hugging Face.")
parser.add_argument("--repo_id", type=str, help="The ID of the repository to upload to.")

args = parser.parse_args()

api = HfApi()
# api.upload_folder(
#     folder_path=f"/mnt/petrelfs/zhuchenglin/LLaVA/checkpoints_ft/{args.repo_id}/llava-v1.5-13b",
#     repo_id=f"starriver030515/{args.repo_id}",
#     repo_type="model",
# )
api.upload_folder(
    folder_path="/mnt/hwfile/zhuchenglin/diffusion",
    repo_id="starriver030515/diffusion",
    repo_type="model",
)
# zip -r /mnt/hwfile/zhuchenglin/LLaVA.zip /mnt/petrelfs/zhuchenglin/LLaVA