Skip to content

How to convert pi05 base model to torch #781

@LukeLIN-web

Description

@LukeLIN-web

Cannot find pi05 base config in https://github.com/Physical-Intelligence/openpi/blob/main/src/openpi/training/config.py ?
Which config should we use

uv run examples/convert_jax_model_to_pytorch.py \
    --checkpoint_dir ./pi05_base \
    --config_name ? \
    --output_path ./pi05_base_pytorch

I tried to use pi05_libero

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions