openpi icon indicating copy to clipboard operation
openpi copied to clipboard

Published 20 hours ago • verified publisher icon Physical-Intelligence

Vectorized inference for pi0 and pi0.5 pytorch version

Open Virlus opened this issue 1 month ago • 0 comments

Hi there. I noticed that the jax version of pi0 does not support vectorized inference (say, batch size > 1). Does the latest pytorch version of these models work?

Virlus avatar Nov 03 '25 04:11 Virlus