UR5 SmolVLA Pick-Cube

UR5 + Robotiq 85 pick-cube policy trained using SmolVLA.

Code: https://github.com/leesweqq/ur5_vla_pick_cube_project

Model

This repository contains:

  • SmolVLA checkpoint
  • Training configuration
  • Policy preprocessors

Usage

from lerobot.policies.smolvla.modeling_smolvla import SmolVLAPolicy

policy = SmolVLAPolicy.from_pretrained(
    "kyle0101/ur5-smolvla-pick-cube-project"
)

Environment

  • Python 3.10
  • LeRobot 0.4.4
  • PyBullet
  • SmolVLA

Task

Pick and lift a red cube using a UR5 robot arm with a Robotiq 85 gripper.

license: mit

Downloads last month
62
Safetensors
Model size
0.5B params
Tensor type
F32
·
BF16
·
Video Preview
loading