mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-16 09:09:48 +00:00
fix(profiling): build flash-attn without isolation for groot
groot depends on flash-attn, which fails to build in uv's default isolated build env because it doesn't declare torch as a build-time dependency. Torch is a core lerobot dep and is already present in the target venv when groot is synced, so we can safely disable build isolation just for flash-attn. The flag is a no-op for policies that don't pull in flash-attn. Made-with: Cursor
This commit is contained in:
@@ -177,6 +177,11 @@ jobs:
|
||||
if [[ -n "${extra}" ]]; then
|
||||
sync_cmd+=(--extra "${extra}")
|
||||
fi
|
||||
# flash-attn does not declare torch as a build-time dep, so its
|
||||
# isolated build env fails with ModuleNotFoundError. Torch is a
|
||||
# core lerobot dep and is already resolved here, so we disable
|
||||
# build isolation for flash-attn specifically.
|
||||
sync_cmd+=(--no-build-isolation-package flash-attn)
|
||||
if ! "${sync_cmd[@]}"; then
|
||||
echo "Dependency install failed for ${policy}; skipping." >&2
|
||||
overall_status=1
|
||||
|
||||
Reference in New Issue
Block a user