pi05-droid-pytorch

PyTorch conversion of Physical Intelligence's official pi05_droid checkpoint (gs://openpi-assets/checkpoints/pi05_droid), converted via openpi examples/convert_jax_model_to_pytorch.py with --config_name pi05_libero, precision float32. Uploaded for RobotTrain subnet validator pipeline testing โ€” a DROID-fine-tuned model is expected to score low on LIBERO, serving as a format-valid negative control (zero-score anchor) for benchmark discrimination tests.

Note: besides the checkpoint's original DROID assets, assets/physical-intelligence/libero/norm_stats.json from the official pi05_libero checkpoint was added so the model can be served with the pi05_libero config (the validator's submission convention).

Important: the LIBERO score of this repo does NOT measure the original checkpoint's capability. The injected LIBERO norm_stats differ from the statistics this model was trained with, and its native state/action spaces do not match LIBERO's (8-dim EEF state / 7-dim delta-EEF actions), so under this pipeline its inputs and outputs are semantically mismatched by construction. Treat it strictly as a format-valid negative control for validator pipeline testing โ€” never as an evaluation of the underlying model.

Layout: model.safetensors (fp32) + assets/<asset_id>/norm_stats.json. Load with openpi: create_trained_policy(get_config("pi05_libero"), <this dir>) (LIBERO convention) or get_config("pi05_droid") with the original DROID stats.

Downloads last month
29
Safetensors
Model size
4B params
Tensor type
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support