Skip to main content

Overview

This document covers the full pipeline of pi0 / pi05 (fine-tuned checkpoint 030000), from LeRobot training through S600 quantization and compilation to on-device deployment and running, as well as the issues encountered at each stage and how to resolve them.

Differences between pi0 and pi05​

Itempi05pi0
version0.50
state injectionno token;
uses AdaRMSNorm (adarms) modulation
state_proj (32→1024) + prefix state token
time injectiontime_mlp → adarms modulationaction_time_mlp concatenated with the action into the input
prompt length20048
action suffix[action_horizon] (50)[state_token] + [action_horizon] (51)
engine time_horizon_01
enable_time_mod_luttruefalse
runtime state injectioninject_state=true (injected into the prompt)inject_state=false (uses the state tensor)