admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-20 19:19:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ci): isolate profiling container home dirs

Browse Source
This commit is contained in:
Pepijn
2026-04-15 22:51:22 +02:00
parent 3ea722c6c0
commit f2d0f04dd0
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/model_profiling.yml
+5 -1
View File
@@ -100,9 +100,12 @@ jobs:
docker run --rm --gpus all \ docker run --rm --gpus all \
--user "$(id -u):$(id -g)" \ --user "$(id -u):$(id -g)" \
--shm-size=16g \ --shm-size=16g \
-e HOME=/tmp/lerobot-home \
-e HF_HOME=/tmp/hf \ -e HF_HOME=/tmp/hf \
-e UV_PROJECT_ENVIRONMENT=/tmp/lerobot-venv \ -e UV_PROJECT_ENVIRONMENT=/tmp/lerobot-venv \
-e UV_CACHE_DIR=/tmp/uv-cache \ -e UV_CACHE_DIR=/tmp/uv-cache \
-e UV_PYTHON_PREFERENCE=only-system \
-e XDG_DATA_HOME=/tmp/xdg-data \
-e XDG_CACHE_HOME=/tmp/xdg-cache \ -e XDG_CACHE_HOME=/tmp/xdg-cache \
-e HOST_GIT_COMMIT="${{ github.event.pull_request.head.sha || github.event.inputs.git_commit || github.sha }}" \ -e HOST_GIT_COMMIT="${{ github.event.pull_request.head.sha || github.event.inputs.git_commit || github.sha }}" \
-e HF_USER_TOKEN="${HF_USER_TOKEN}" \ -e HF_USER_TOKEN="${HF_USER_TOKEN}" \
@@ -114,8 +117,9 @@ jobs:
-v "${GITHUB_WORKSPACE}:/workspace" \ -v "${GITHUB_WORKSPACE}:/workspace" \
-w /workspace \ -w /workspace \
huggingface/lerobot-gpu:latest \ huggingface/lerobot-gpu:latest \
bash -lc ' bash -c '
set -euxo pipefail set -euxo pipefail
mkdir -p "${HOME}" "${HF_HOME}" "${UV_CACHE_DIR}" "${XDG_CACHE_HOME}" "${XDG_DATA_HOME}"
rm -rf /tmp/lerobot-src rm -rf /tmp/lerobot-src
cp -a /workspace/. /tmp/lerobot-src cp -a /workspace/. /tmp/lerobot-src
cd /tmp/lerobot-src cd /tmp/lerobot-src