Zhengyiluo commited on
Commit
3c82700
Β·
verified Β·
1 Parent(s): 7c90a56

Rename checkpoint release to sonic_v1_1

Browse files
README.md CHANGED
@@ -145,7 +145,7 @@ variant.
145
  |---|---|---|---|
146
  | **Default SONIC (original release)** | Top-level deployment files; PyTorch checkpoint under `sonic_release/` | 10 frames at 20 ms spacing, approximately 200 ms | General-purpose motion tracking, planning, and teleoperation. G1 and teleoperation references use `step5`. |
147
  | **Low-latency teleoperation** | `low_latency/` | 4 frames at 20 ms spacing, approximately 80 ms | Reduced-reference-lookahead controller for responsive teleoperation and VLA execution. G1 and teleoperation references use `step1`. |
148
- | **Heading/wrist teleoperation** | `heading_wrist/` | 10 frames at 20 ms spacing, approximately 200 ms | Robot-heading-normalized controller trained with wrist-pose augmentation for heading-stable 3-point teleoperation and SONIC-backed VLA policies. G1 and teleoperation references use `step5`; this is not the low-latency model. |
149
 
150
  All variants use 64-dimensional universal motion tokens and a 29-DoF action
151
  decoder. Inference is powered by TensorRT and runs on both desktop and Jetson
@@ -158,8 +158,8 @@ hardware.
158
  | Kinematic planner | `planner_sonic.onnx` | Real-time locomotion style planner |
159
  | Low-latency policy | `low_latency/model_encoder.onnx`, `low_latency/model_decoder.onnx`, `low_latency/observation_config.yaml` | Reduced-lookahead G1 controller variant |
160
  | Low-latency PyTorch checkpoint | `low_latency/last.pt` | Training checkpoint and config for the low-latency variant |
161
- | Heading/wrist policy | `heading_wrist/model_encoder.onnx`, `heading_wrist/model_decoder.onnx`, `heading_wrist/observation_config.yaml` | Heading-normalized G1 controller variant |
162
- | Heading/wrist PyTorch checkpoint | `heading_wrist/last.pt` | Training checkpoint and configs for the heading/wrist variant |
163
 
164
  **Quick download** (requires `pip install huggingface_hub`):
165
 
@@ -174,14 +174,14 @@ Or use the download script from the GitHub repo:
174
  python download_from_hf.py # policy + planner (default)
175
  python download_from_hf.py --no-planner # policy only
176
  python download_from_hf.py --low-latency # low-latency policy + planner
177
- python download_from_hf.py --heading-wrist # heading/wrist policy + planner
178
  ```
179
 
180
  See the [Download Models guide](https://nvlabs.github.io/GR00T-WholeBodyControl/getting_started/download_models.html) for full instructions.
181
 
182
- ## Heading/Wrist Inference
183
 
184
- The `heading_wrist/` variant uses robot-heading-normalized target orientation
185
  and a 10-frame SMPL/wrist reference horizon. It was trained with wrist-pose
186
  augmentation and is intended for 3-point teleoperation and SONIC-backed VLA
187
  policies trained against this controller.
@@ -190,12 +190,12 @@ policies trained against this controller.
190
  git clone https://github.com/NVlabs/GR00T-WholeBodyControl.git
191
  cd GR00T-WholeBodyControl
192
  pip install huggingface_hub
193
- python download_from_hf.py --heading-wrist
194
 
195
  cd gear_sonic_deploy
196
  ./deploy.sh \
197
- --cp policy/heading_wrist/model \
198
- --obs-config policy/heading_wrist/observation_config.yaml \
199
  sim
200
  ```
201
 
 
145
  |---|---|---|---|
146
  | **Default SONIC (original release)** | Top-level deployment files; PyTorch checkpoint under `sonic_release/` | 10 frames at 20 ms spacing, approximately 200 ms | General-purpose motion tracking, planning, and teleoperation. G1 and teleoperation references use `step5`. |
147
  | **Low-latency teleoperation** | `low_latency/` | 4 frames at 20 ms spacing, approximately 80 ms | Reduced-reference-lookahead controller for responsive teleoperation and VLA execution. G1 and teleoperation references use `step1`. |
148
+ | **SONIC v1.1** | `sonic_v1_1/` | 10 frames at 20 ms spacing, approximately 200 ms | Robot-heading-normalized controller trained with wrist-pose augmentation for heading-stable 3-point teleoperation and SONIC-backed VLA policies. G1 and teleoperation references use `step5`; this is not the low-latency model. |
149
 
150
  All variants use 64-dimensional universal motion tokens and a 29-DoF action
151
  decoder. Inference is powered by TensorRT and runs on both desktop and Jetson
 
158
  | Kinematic planner | `planner_sonic.onnx` | Real-time locomotion style planner |
159
  | Low-latency policy | `low_latency/model_encoder.onnx`, `low_latency/model_decoder.onnx`, `low_latency/observation_config.yaml` | Reduced-lookahead G1 controller variant |
160
  | Low-latency PyTorch checkpoint | `low_latency/last.pt` | Training checkpoint and config for the low-latency variant |
161
+ | SONIC v1.1 policy | `sonic_v1_1/model_encoder.onnx`, `sonic_v1_1/model_decoder.onnx`, `sonic_v1_1/observation_config.yaml` | Robot-heading-normalized G1 controller variant |
162
+ | SONIC v1.1 PyTorch checkpoint | `sonic_v1_1/last.pt` | Training checkpoint and configs for SONIC v1.1 |
163
 
164
  **Quick download** (requires `pip install huggingface_hub`):
165
 
 
174
  python download_from_hf.py # policy + planner (default)
175
  python download_from_hf.py --no-planner # policy only
176
  python download_from_hf.py --low-latency # low-latency policy + planner
177
+ python download_from_hf.py --sonic-v1-1 # SONIC v1.1 policy + planner
178
  ```
179
 
180
  See the [Download Models guide](https://nvlabs.github.io/GR00T-WholeBodyControl/getting_started/download_models.html) for full instructions.
181
 
182
+ ## SONIC v1.1 Inference
183
 
184
+ The `sonic_v1_1/` variant uses robot-heading-normalized target orientation
185
  and a 10-frame SMPL/wrist reference horizon. It was trained with wrist-pose
186
  augmentation and is intended for 3-point teleoperation and SONIC-backed VLA
187
  policies trained against this controller.
 
190
  git clone https://github.com/NVlabs/GR00T-WholeBodyControl.git
191
  cd GR00T-WholeBodyControl
192
  pip install huggingface_hub
193
+ python download_from_hf.py --sonic-v1-1
194
 
195
  cd gear_sonic_deploy
196
  ./deploy.sh \
197
+ --cp policy/sonic_v1_1/model \
198
+ --obs-config policy/sonic_v1_1/observation_config.yaml \
199
  sim
200
  ```
201
 
{heading_wrist β†’ sonic_v1_1}/config.yaml RENAMED
File without changes
{heading_wrist β†’ sonic_v1_1}/last.pt RENAMED
File without changes
{heading_wrist β†’ sonic_v1_1}/model_config.yaml RENAMED
File without changes
{heading_wrist β†’ sonic_v1_1}/model_decoder.onnx RENAMED
File without changes
{heading_wrist β†’ sonic_v1_1}/model_encoder.onnx RENAMED
File without changes
{heading_wrist β†’ sonic_v1_1}/observation_config.yaml RENAMED
File without changes