diff --git a/docker/Dockerfile.benchmark.libero b/docker/Dockerfile.benchmark.libero index e13cf7de1..b3969d491 100644 --- a/docker/Dockerfile.benchmark.libero +++ b/docker/Dockerfile.benchmark.libero @@ -65,7 +65,7 @@ RUN uv venv --python python${PYTHON_VERSION} COPY --chown=user_lerobot:user_lerobot setup.py pyproject.toml uv.lock README.md MANIFEST.in ./ COPY --chown=user_lerobot:user_lerobot src/ src/ -RUN uv sync --locked --extra libero --no-cache +RUN uv sync --locked --extra libero --extra smolvla --no-cache # Pre-create libero's config file pointing to the bundled package assets. # libero/libero/__init__.py calls input() when ~/.libero/config.yaml is missing. diff --git a/docker/Dockerfile.benchmark.metaworld b/docker/Dockerfile.benchmark.metaworld index f89036a71..0c916c553 100644 --- a/docker/Dockerfile.benchmark.metaworld +++ b/docker/Dockerfile.benchmark.metaworld @@ -65,7 +65,7 @@ RUN uv venv --python python${PYTHON_VERSION} COPY --chown=user_lerobot:user_lerobot setup.py pyproject.toml uv.lock README.md MANIFEST.in ./ COPY --chown=user_lerobot:user_lerobot src/ src/ -RUN uv sync --locked --extra metaworld --no-cache +RUN uv sync --locked --extra metaworld --extra smolvla --no-cache RUN chmod +x /lerobot/.venv/lib/python${PYTHON_VERSION}/site-packages/triton/backends/nvidia/bin/ptxas