pi0.5 / requirements.txt
XiangpengYang's picture
fix: add OpenPI logging progress dependency
8dc20ae
Raw
History Blame Contribute Delete
645 Bytes
gradio==5.49.1
huggingface-hub==0.36.2
pandas==2.2.3
pillow==11.1.0
numpy==1.26.4
augmax>=0.3.4
dm-tree>=0.1.8
einops>=0.8.0
equinox>=0.11.8
flax==0.10.2
fsspec[gcs]>=2024.6.0
jax[cuda12]==0.5.3
jaxtyping==0.2.36
ml-collections==1.0.0
numpydantic>=1.6.6
opencv-python-headless>=4.10.0.84
orbax-checkpoint==0.11.13
sentencepiece>=0.2.0
torch==2.11.0
typing-extensions>=4.12.2
tyro>=0.9.5
tqdm-loggable>=0.2
wandb>=0.19.1
filelock>=3.16.1
beartype==0.19.0
chex==0.1.90
treescope>=0.1.7
transformers==4.53.2
rich>=14.0.0
polars>=1.30.0
pytest==8.3.5
lerobot @ git+https://github.com/huggingface/lerobot.git@0cf864870cf29f4738d3ade893e6fd13fbd7cdb5