Skip to content

Add GR00T N1.7 training and rollout support - #64

Merged
nicolas-rabault merged 6 commits into
huggingface:mainfrom
EAOZONE:main
Aug 3, 2026
Merged

Add GR00T N1.7 training and rollout support#64
nicolas-rabault merged 6 commits into
huggingface:mainfrom
EAOZONE:main

Conversation

@EAOZONE

@EAOZONE EAOZONE commented Jul 12, 2026

Copy link
Copy Markdown
Contributor

Summary

This PR adds training and rollout support for NVIDIA GR00T N1.7.

  • Add GR00T policy configuration and dependency preflight support.
  • Expose GR00T-specific training options in the frontend.
  • Include the rebuilt frontend bundle.
  • Use RTC chunked inference for GR00T and other relative-action policies.
  • Configure the RTC execution horizon from the policy’s action-step count.
  • Set the inference queue threshold to zero for stable chunk execution.
  • Provide a friendly error message when relative-action policies are run without RTC.
  • Add test coverage for relative-action rollout configuration.

Validation

  • pytest -q tests/test_train.py tests/test_rollout.py — 34 passed
  • ESLint on all changed frontend source files — passed
  • npm run build — passed
  • git diff --check — passed
  • Successfully trained a GR00T policy and ran it on a physical SO-101 robot.

Notes

The repository-wide frontend lint reports pre-existing errors in unrelated files. All frontend files changed by this PR pass ESLint.

@nicolas-rabault
nicolas-rabault self-requested a review July 16, 2026 13:21

@nicolas-rabault nicolas-rabault left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nice feature, verified all the GR00T/RTC flags against lerobot v0.6.0, they're correct.

Drop the committed frontend/dist/; CI (build_frontend.yml) rebuilds it on main; it just bloats the diff (~1.4MB) and will conflict.

Comment thread lelab/train.py Outdated
Comment thread lelab/train.py
Comment thread lelab/rollout.py Outdated
Comment thread frontend/src/components/training/config/GrootCard.tsx
@EAOZONE
EAOZONE force-pushed the main branch 3 times, most recently from d5ad370 to 8fe8bac Compare July 16, 2026 14:06
@EAOZONE
EAOZONE requested a review from nicolas-rabault July 16, 2026 17:41
nicolas-rabault
nicolas-rabault previously approved these changes Aug 3, 2026

@nicolas-rabault nicolas-rabault left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for the thorough follow-up,.

@nicolas-rabault nicolas-rabault left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Heads up, I rebased your branch onto main and force-pushed. Your six commits are intact, the only changes were resolving conflicts in the committed frontend/dist bundle (all resolved to main's version, so the branch no longer touches dist at all) and one commit wrapping two lines over the ruff limit.

@nicolas-rabault
nicolas-rabault merged commit 39fcfee into huggingface:main Aug 3, 2026
2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants