Skip to content

add policy client for GR00T N1.6 - #104

Merged
EverNorif merged 3 commits into
LightwheelAI:mainfrom
BotRunner64:feat/issue-103
Jan 16, 2026
Merged

EverNorif merged 3 commits into
LightwheelAI:mainfrom
BotRunner64:feat/issue-103

Conversation

@BotRunner64

Copy link
Copy Markdown
Contributor

Summary:
add Gr00t16ServicePolicyClient to support GR00t N1.6 new policy server api #103 , compatibility with other policy types
, use as:

python scripts/evaluation/policy_inference.py \
    --task=LeIsaac-SO101-PickOrange-v0 \
    --eval_rounds=10 \
    --policy_type=gr00tn1.6 \
    --policy_host=localhost \
    --policy_port=5555 \
    --policy_timeout_ms=5000 \
    --policy_action_horizon=16 \
    --policy_language_instruction="Pick up the orange and place it on the plate" \
    --device=cuda \
    --enable_cameras

Verification:
Tested locally: Confirmed successful deploy of finetuned GR00T N1.6 model

@EverNorif
EverNorif self-requested a review January 4, 2026 06:45
@EverNorif

EverNorif commented Jan 4, 2026 •

Copy link
Copy Markdown
Collaborator

@12e21 Thank you for your contribution.
I don’t have the resources to test the GR00T v1.6 related changes this week, will carry out the testing after one week. :)

@EverNorif

Copy link
Copy Markdown
Collaborator

@12e21 Hi, I’m currently testing this PR, but I ran into an issue when fine-tuning gr00t n1.6. I’m fine-tuning on a 4090, but even when I set global_batch_size = 1, I still get a CUDA Out of Memory error. Could you share your fine-tuning configuration?

@BotRunner64

Copy link
Copy Markdown
Contributor Author

@EverNorif I tried finetuning with 4090 24GB, indeed OOM with batch size 1; on A100 or A6000, it works with about 40GB VRAM

@BotRunner64

Copy link
Copy Markdown
Contributor Author

@EverNorif I can share my config and upload finetuned policy model to huggingface, so you can test it

@EverNorif

Copy link
Copy Markdown
Collaborator

@EverNorif I can share my config and upload finetuned policy model to huggingface, so you can test it

Okay, that would be better.

@BotRunner64

Copy link
Copy Markdown
Contributor Author

@EverNorif I have upload configs, scripts, and checkpoint, find at 12e21/gr00t_n1d6_leisaac_pick_orange

@EverNorif
EverNorif merged commit 6b933e8 into LightwheelAI:main Jan 16, 2026
2 checks passed
@EverNorif EverNorif linked an issue Jan 16, 2026 that may be closed by this pull request
harsh-mulodhia pushed a commit to harsh-mulodhia/leisaac that referenced this pull request Jun 24, 2026
* add policy client for GR00T N1.6

* add target commit for gr00tn1.6.

* add related docs for policy inference of gr00tn1.6.

---------

Co-authored-by: EverNorif <1320346985@qq.com>
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.

add support for GR00T N1.6

2 participants