fix(docker): install VLABench from GitHub, not PyPI

VLABench is not published on PyPI. Install from the OpenMOSS/VLABench
GitHub repo directly.

Made-with: Cursor
This commit is contained in:
Pepijn
2026-04-16 17:04:09 +02:00
parent 1a0aaf64f4
commit 8053f2eea0
+4 -3
View File
@@ -20,9 +20,10 @@
FROM huggingface/lerobot-gpu:latest FROM huggingface/lerobot-gpu:latest
# Install VLABench and pin MuJoCo/dm-control versions. # Install VLABench from GitHub (not on PyPI) and pin MuJoCo/dm-control.
# The base image uses uv, not pip. RUN uv pip install --no-cache \
RUN uv pip install --no-cache vlabench mujoco==3.2.2 dm-control==1.0.22 && \ "vlabench @ git+https://github.com/OpenMOSS/VLABench.git" \
mujoco==3.2.2 dm-control==1.0.22 && \
python -c "from VLABench.utils import download_assets; download_assets()" python -c "from VLABench.utils import download_assets; download_assets()"
# Overlay the PR's source code on top of the nightly image. # Overlay the PR's source code on top of the nightly image.