Downloads · 30 days
0
magiclab01/pi0_Magicbot_PnP_fruits
pi0_Magicbot_PnP_fruits is a machine learning model from magiclab01. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
The model weights are stored in the 80000 subdirectory of the repository, which is a pi0 Orbax checkpoint used for inference on Magicbot Gen1 robot. Please refer to magicbot-gen1pi0demo project for detailed information.
Downloads · 30 days
0
Access
Public
Updated Aug 25, 2025
Repo size
6.2 GB
Likes
0
Public
Click a slice to open those files.
Other6.2 GB · 100%
From the Hugging Face model README
The model weights are stored in the 80000 subdirectory of the repository, which is a pi0 Orbax checkpoint used for inference on Magicbot Gen1 robot.
Please refer to magicbot-gen1_pi0_demo project for detailed information.
Use the following code to load the model:
from huggingface_hub import snapshot_download
import orbax.checkpoint
import os
# Download the repository
checkpoint_dir = snapshot_download(repo_id="magiclab01/pi0-Magicbot-PnP-fruits")
checkpoint_subdir = os.path.join(checkpoint_dir, "80000")
# Load Orbax checkpoint
checkpointer = orbax.checkpoint.PyTreeCheckpointer()
state_dict = checkpointer.restore(checkpoint_subdir)