lerobot icon indicating copy to clipboard operation
lerobot copied to clipboard

Convert PI0Fast model to torch model

Open Liuzeyi25 opened this issue 6 months ago • 1 comments

hi, I want to ask how to convert pi0fast Jax model to torch model, the codebase seems like not provide a script, thank you very much!!

Liuzeyi25 avatar Jun 09 '25 10:06 Liuzeyi25

same here

cyteena avatar Jun 09 '25 10:06 cyteena

Hi we update our pi0 implementation and released pi05 support in this PR: https://github.com/huggingface/lerobot/pull/1910 🚀 . For the FAST implementation the openpi repo has a script to convert their weights to PyTorch

pkooij avatar Oct 08 '25 08:10 pkooij