keycharon commited on
Commit
e3a6d27
·
verified ·
1 Parent(s): 2bf51a1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +355 -0
README.md CHANGED
@@ -1,3 +1,358 @@
1
  ---
2
  license: apache-2.0
3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
 
4
+ tags:
5
+ - humanoid-robotics
6
+ - loco-manipulation
7
+ - world-action-model
8
+ - embodied-ai
9
+ - robot-learning
10
+ - vision-language-action
11
+ - diffusion
12
+ - latent-world-model
13
+ - unitree-g1
14
+ ---
15
+
16
+ # ω-0: A Latent Predictive World Action Model for Concurrent Humanoid Loco-Manipulation
17
+
18
+ **ω-0 (OMEGA-0)** is a latent predictive world action model for real-world humanoid **concurrent loco-manipulation**.
19
+
20
+ Given a language instruction, visual observations, and robot proprioceptive state, ω-0 predicts **controller-compatible whole-body action latents** for humanoid robot execution. Instead of reconstructing future RGB videos in pixel space, ω-0 predicts compact future visual representations and couples them with diffusion-based whole-body action generation.
21
+
22
+ The released checkpoint is intended to be used with the official OMEGA-0 codebase for inference and real-robot deployment.
23
+
24
+ > **Paper:** *ω-0: A Latent Predictive World Action Model for Concurrent Humanoid Loco-Manipulation*
25
+ > **Code:** https://github.com/gentlefress/OMEGA-0
26
+ > **Project Page:** https://gentlefress.github.io/OMEGA-0_page/
27
+ > **Dataset:** https://huggingface.co/datasets/keycharon/omega-HOME
28
+
29
+ ---
30
+
31
+ ## 🌟 Highlights
32
+
33
+ - **Whole-body loco-manipulation:** jointly coordinates locomotion, posture, balance, and manipulation.
34
+ - **Latent predictive modeling:** predicts future visual representations rather than reconstructing future frames in pixel space.
35
+ - **Action generation:** generates whole-body action latents compatible with the SONIC controller.
36
+ - **Multimodal conditioning:** supports language, visual observations, and robot state.
37
+ - **Real-world humanoid deployment:** demonstrated on the Unitree G1.
38
+ - **Multiple visual viewpoints:** supports egocentric RGB, exocentric RGB, and exocentric depth.
39
+ - **Single unified policy:** one model is used across a diverse set of household loco-manipulation tasks.
40
+ - **Open-source workflow:** the official repository provides training, inference, teleoperation, recording, and deployment code.
41
+
42
+ ---
43
+
44
+ ## 🧠 Model Overview
45
+
46
+ ω-0 follows a simple principle:
47
+
48
+ > **Predict how the scene will evolve, while directly generating how the whole body should act.**
49
+
50
+ The model combines three complementary components:
51
+
52
+ ```text
53
+ Language Instruction
54
+ │
55
+ ├──────────────┐
56
+ │ │
57
+ Visual Observation Robot State
58
+ │ │
59
+ └──────┬───────┘
60
+ ↓
61
+ Multimodal Encoder
62
+ │
63
+ ↓
64
+ ┌─────────────────────┐
65
+ │ ω-0 WAM Predictor │
66
+ └─────────────────────┘
67
+ │ │
68
+ │ └────────→ Future Visual Latent
69
+ │
70
+ └────────────────────→ Whole-body Action Latent
71
+ │
72
+ ↓
73
+ SONIC
74
+ │
75
+ ↓
76
+ Unitree G1
77
+ ```
78
+
79
+ The predictive branch provides a compact representation of future scene evolution, while the action branch generates controller-compatible whole-body actions.
80
+
81
+ ---
82
+
83
+ ## 📥 Inputs
84
+
85
+ Depending on the configured model and deployment setup, ω-0 can use:
86
+
87
+ ### Language
88
+
89
+ A natural-language task instruction, for example:
90
+
91
+ ```text
92
+ "Pick up the clothes from the washing machine."
93
+ ```
94
+
95
+ ### Visual Observation
96
+
97
+ Supported observation modes include:
98
+
99
+ - Egocentric RGB
100
+ - Exocentric RGB
101
+ - Exocentric depth
102
+
103
+ ### Robot Proprioception
104
+
105
+ Robot state information can be used as specified by the corresponding configuration.
106
+
107
+ ---
108
+
109
+ ## 📤 Outputs
110
+
111
+ The model predicts **whole-body action latents**, which are subsequently decoded/interpreted by the OMEGA-0 action pipeline and executed through the **SONIC** whole-body controller.
112
+
113
+ The checkpoint therefore should **not** be interpreted as a standalone joint-position predictor.
114
+
115
+ The complete execution pipeline is:
116
+
117
+ ```text
118
+ Instruction + Observation + Robot State
119
+ ↓
120
+ ω-0
121
+ ↓
122
+ Whole-body Action Latent
123
+ ↓
124
+ Action Decoder
125
+ ↓
126
+ SONIC
127
+ ↓
128
+ Unitree G1
129
+ ```
130
+
131
+ ---
132
+
133
+ ## 🤖 Real-World Platform
134
+
135
+ The released system targets the following experimental setup:
136
+
137
+ - **Robot:** Unitree G1
138
+ - **Hands:** Inspire dexterous hands
139
+ - **Controller:** SONIC
140
+ - **Egocentric camera:** ZED Mini
141
+ - **Teleoperation:** PICO body tracking through XRoboToolkit
142
+
143
+ Hardware and software configurations may need to be adapted to your own setup.
144
+
145
+ ---
146
+
147
+ ## 🚀 Quick Start
148
+
149
+ ### 1. Clone the official repository
150
+
151
+ ```bash
152
+ git clone https://github.com/gentlefress/OMEGA-0.git
153
+ cd OMEGA-0
154
+ ```
155
+
156
+ ### 2. Install the environment
157
+
158
+ The official repository uses `uv`:
159
+
160
+ ```bash
161
+ uv sync --locked --extra train --extra deploy --python 3.10 --inexact
162
+ source .venv/bin/activate
163
+
164
+ uv pip install flash-attn --no-build-isolation
165
+ ```
166
+
167
+ Verify the installation:
168
+
169
+ ```bash
170
+ python -c "import omega, torch; print('PyTorch:', torch.__version__); print('CUDA:', torch.cuda.is_available())"
171
+ ```
172
+
173
+ ### 3. Download the checkpoint
174
+
175
+ Download the checkpoint from this Hugging Face repository and place it in your local model directory.
176
+
177
+ For example:
178
+
179
+ ```text
180
+ /path/to/omega0_checkpoint/
181
+ ```
182
+
183
+ ### 4. Configure the inference server
184
+
185
+ The official OMEGA-0 repository provides an inference configuration under:
186
+
187
+ ```text
188
+ src/configs/serve.yaml
189
+ ```
190
+
191
+ For an exported checkpoint, configure the model artifact:
192
+
193
+ ```yaml
194
+ model:
195
+ artifact: /path/to/omega0_checkpoint
196
+ device: cuda
197
+ ```
198
+
199
+ Then validate the configuration:
200
+
201
+ ```bash
202
+ python -m omega.inference.run --config src/configs/serve.yaml --check
203
+ ```
204
+
205
+ Launch the inference server:
206
+
207
+ ```bash
208
+ OPENBLAS_NUM_THREADS=1 OMP_NUM_THREADS=4 \
209
+ HF_HUB_OFFLINE=1 TRANSFORMERS_OFFLINE=1 \
210
+ python -m omega.inference.run --config src/configs/serve.yaml
211
+ ```
212
+
213
+ The default inference server listens on:
214
+
215
+ ```text
216
+ 127.0.0.1:8014
217
+ ```
218
+
219
+ For multi-machine deployment, configure the server and client addresses according to your network setup.
220
+
221
+ ---
222
+
223
+ ## 🦾 Robot Deployment
224
+
225
+ The official repository provides a robot client for Unitree G1 deployment.
226
+
227
+ Set the inference endpoint and task instruction:
228
+
229
+ ```bash
230
+ export INFERENCE_ENDPOINT=http://127.0.0.1:8014
231
+ export INSTRUCTION="Pick up the clothes from the washing machine."
232
+ ```
233
+
234
+ Then run:
235
+
236
+ ```bash
237
+ python -m omega_real.run --config real/configs/deploy.yaml --check
238
+ python -m omega_real.run --config real/configs/deploy.yaml --arm
239
+ ```
240
+
241
+ Please follow the official deployment instructions and safety procedures before enabling robot commands.
242
+
243
+ ---
244
+
245
+ ## 🏋️ Training
246
+
247
+ The released codebase contains two main training stages:
248
+
249
+ ### Stage 1 — Action-Token Pretraining
250
+
251
+ The model first learns to associate language and visual observations with whole-body motion tokens.
252
+
253
+ ```text
254
+ Language + Visual Observation
255
+ ↓
256
+ Qwen3-VL
257
+ ↓
258
+ Whole-body motion tokens
259
+ ```
260
+
261
+ ### Stage 2 — WAM Fine-Tuning
262
+
263
+ The model is further optimized to jointly predict future visual representations and whole-body action latents.
264
+
265
+ ```text
266
+ Language + Observation + State
267
+ ↓
268
+ ω-0 WAM
269
+ ↙ ↘
270
+ Future Visual Action Latent
271
+ Representation
272
+ ↓
273
+ SONIC
274
+ ```
275
+
276
+ Training and dataset preparation instructions are available in the official repository.
277
+
278
+ ---
279
+
280
+ ## 📊 ω-HOME Dataset
281
+
282
+ ω-0 is accompanied by **ω-HOME**, a 40+ hour real-world household humanoid dataset containing synchronized multimodal observations, whole-body motion, robot states, and action latents.
283
+
284
+ The dataset is available on Hugging Face:
285
+
286
+ https://huggingface.co/datasets/keycharon/omega-HOME
287
+
288
+ ---
289
+
290
+ ## 🎯 Intended Use
291
+
292
+ This checkpoint is intended for:
293
+
294
+ - Research on humanoid robot loco-manipulation
295
+ - World action models
296
+ - Whole-body robot learning
297
+ - Vision-language-action modeling
298
+ - Latent predictive modeling
299
+ - Real-world humanoid manipulation
300
+ - Reproduction and extension of the ω-0 experiments
301
+
302
+ The checkpoint is primarily a **research model** and is not intended for unsupervised deployment in safety-critical environments.
303
+
304
+ ---
305
+
306
+ ## ⚠️ Limitations and Safety
307
+
308
+ ω-0 is a research system and should be deployed with appropriate hardware-level and software-level safety mechanisms.
309
+
310
+ In particular:
311
+
312
+ - Real-robot behavior depends on the underlying robot, controller, camera, calibration, and environment.
313
+ - The released model is developed and evaluated primarily on the Unitree G1 platform.
314
+ - Performance may degrade under substantial distribution shifts in scenes, objects, embodiments, or tasks.
315
+ - Generated actions should be checked by the downstream controller and deployment stack.
316
+ - Do not operate the robot near people without appropriate safety precautions.
317
+ - The model should not be treated as a guaranteed safe or autonomous control policy.
318
+
319
+ ---
320
+
321
+ ## 📚 Citation
322
+
323
+ If you use ω-0 or this checkpoint in your research, please cite:
324
+
325
+ ```bibtex
326
+ @article{li2026omega0,
327
+ title = {{$\omega$-0: A Latent Predictive World Action Model for Concurrent Humanoid Loco-Manipulation}},
328
+ author = {Zhe Li and Zhenzhe Zhang and Yangyang Wei and Wenjie Zhang and
329
+ Xichen Yuan and Peiyuan Zhi and Gen Li and Xinying Guo and
330
+ Fengjie Gao and Jianfei Yang and Shanghang Zhang},
331
+ journal = {arXiv preprint arXiv:2608.06375},
332
+ year = {2026},
333
+ doi = {10.48550/arXiv.2608.06375},
334
+ url = {https://arxiv.org/abs/2608.06375}
335
+ }
336
+ ```
337
+
338
+ ---
339
+
340
+ ## 🙏 Acknowledgements
341
+
342
+ We acknowledge the open-source projects that contributed to the development of OMEGA-0, including:
343
+
344
+ - Ψ₀
345
+ - GR00T-WholeBodyControl
346
+ - Qwen3-VL
347
+ - V-JEPA
348
+ - Wan
349
+
350
+ Please refer to the official repository and the corresponding third-party licenses for the exact dependencies and attribution requirements.
351
+
352
+ ---
353
+
354
+ ## 📄 License
355
+
356
+ OMEGA-0 code is released under the **Apache 2.0 License**. Third-party code, models, and assets remain subject to their respective licenses.
357
+
358
+ Please check the license terms of all external components before redistribution or commercial use.