HuggingFace에 모델 업로드(선택 사항)

모델 Repo 생성

모델 Repo 확인

https://huggingface.co/TommyZihao/lerobot_zihao_model_a

현재는 비어 있습니다

모델 업로드

upload_model.py 파일을 생성하고, 내용은 다음과 같습니다

from huggingface_hub import HfApi

api = HfApi()

repo_id = "TommyZihao/lerobot_zihao_model_shake_hands"

api.upload_folder(
    folder_path="~/output_lerobot_train/b/checkpoints/last/pretrained_model",
    repo_id=repo_id,
    repo_type="model"
)

api.create_tag(repo_id, tag="v0.1.0", repo_type="model")

실행

python upload_model.py

모델 Repo 확인

https://huggingface.co/TommyZihao/lerobot_zihao_model_a

이제 모델 파일이 있습니다