Téléverser le modèle sur HuggingFace (facultatif)

Créer un Model Repo

Consulter le Model Repo

https://huggingface.co/TommyZihao/lerobot_zihao_model_a

Il est vide pour l'instant

Téléverser le modèle

Créez le fichier upload_model.py avec le contenu suivant

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")

Exécuter

python upload_model.py

Consulter le Model Repo

https://huggingface.co/TommyZihao/lerobot_zihao_model_a

Le modèle est maintenant présent