Downloads · 30 days
12
9% of all-time downloads
jafermarq/lerobot123
lerobot123 is a robotics model from jafermarq. Use it for the robotics task on the model card, and read the license before you ship it in a product. It is set up for lerobot.
This model has been pushed to the Hub using the PytorchModelHubMixin integration: - Library: https://github.com/huggingface/lerobot - Docs: [More Information Needed]
Downloads · 30 days
12
9% of all-time downloads
All-time downloads
132
Public
Parameters
76.1M
304 MB on disk
Likes
1
Public
Click a slice to open those files.
.safetensors304 MB · 100%
From the Hugging Face model README
This model has been pushed to the Hub using the PytorchModelHubMixin integration:
The checkpoint in this repository is obtained by running the Flower x LeRobot quickstart-example. This is the global model at round 35 which seemed to give good results.
[!NOTE] Install the version of LeRobot used by the example (the latest commit at the time the example was put together, mid Dec'24). You might want to try using a more recente version.
pip install "lerobot[pusht] @ git+https://github.com/huggingface/lerobot.git@96c7052777aca85d4e55dfba8f81586103ba8f61"
from lerobot.common.policies.diffusion.configuration_diffusion import DiffusionConfig
from lerobot.common.policies.diffusion.modeling_diffusion import DiffusionPolicy
from lerobot.common.datasets.lerobot_dataset import LeRobotDataset
dataset = LeRobotDataset("lerobot/pusht")
cfg = DiffusionConfig(down_dims=[256, 512, 1024])
policy = DiffusionPolicy(cfg, dataset_stats=dataset.stats)
policy.from_pretrained("jafermarq/lerobot123")
print(policy)