Robotics
LeRobot
Safetensors
smolvla
smolvla_base / README.md
nielsr's picture
nielsr HF Staff
Add library name and link to code
cd9e12c verified
|
raw
history blame
1.15 kB
metadata
pipeline_tag: robotics
tags:
  - lerobot
library_name: lerobot

SmolVLA: A vision-language-action model for affordable and efficient robotics

Paper

Code

Designed by Hugging Face.

This model has 450M parameters in total. You can use inside the LeRobot library.

Install smolvla extra dependencies:

pip install -e ".[smolvla]"

Example of finetuning the smolvla pretrained model (smolvla_base):

python lerobot/scripts/train.py \
--policy.path=lerobot/smolvla_base \
--dataset.repo_id=danaaubakirova/svla_so100_task1_v3 \
--batch_size=64 \
--steps=200000

Example of finetuning the smolvla neural network with pretrained VLM and action expert intialized from scratch:

python lerobot/scripts/train.py \
--policy.type=smolvla \
--dataset.repo_id=danaaubakirova/svla_so100_task1_v3 \
--batch_size=64 \
--steps=200000

Example of using the smolvla pretrained model outside LeRobot training framework:

policy = SmolVLAPolicy.from_pretrained("lerobot/smolvla_base")