Instructions to use hchen657Naza/PWM with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use hchen657Naza/PWM with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("hchen657Naza/PWM", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
|
Download PWM_code/RISE_aux/README.md from hchen657Naza/PWM: direct link, hf CLI and curl.
- Browser
- Download file 3.3 kB
-
https://huggingface.co/hchen657Naza/PWM/resolve/main/PWM_code/RISE_aux/README.md
- Command line
-
hf download hf://hchen657Naza/PWM/PWM_code/RISE_aux/README.md
-
curl -L -o README.md https://huggingface.co/hchen657Naza/PWM/resolve/main/PWM_code/RISE_aux/README.md
3.3 kB
๐ RISE
๐ฅ Highlights
- A compositional world model. A principled design that combines a controllable multi-view dynamics model with a progress value model, yielding informative advantages for robust policy improvement.
- RL in imagination. A scalable self-improving framework that bootstraps robot policies through imaginary rollouts, avoiding the hardware cost and laborious reset of real-world interactions.
- Real-world manipulation gains. Non-trivial performance improvements on challenging dexterous tasks, including +35% on dynamic brick sorting, +45% on backpack packing, and +35% on box closing.
๐บ๏ธ Overview
RISE repository is structured with three major parts:
policy_and_value/policy_offline_and_value: OpenPI-based offline policy and value training. They are put together since they share similar model architecture and training pipeline.dynamics/dynamics_model: Action-conditioned dynamics model.policy_and_value/policy_online: Online RL for policy improvement.
๐งโ๐ป Getting Started
- ๐ป Installation
- ๐ Offline Policy Training & Value Model
- ๐ฎ Dynamics Model
- ๐ฆพ Online Policy Training
- ๐ ๏ธ Deployment on Piper
โค๏ธ Acknowledgement
We thank the following projects for their open-source contributions: OpenPI Pi0 / Pi05, Genie Envisioner, RLinf, LTX-Video.
๐ข News
- [2026/04/22] Training code and pre-trained dynamics model are released.
- [2026/02/11] Paper released on arXiv.
๐ License and Citation
All assets and code in this repository are under the Apache 2.0 license unless specified otherwise. The data and checkpoint are under CC BY-NC-SA 4.0. Other modules inherit their own distribution licenses.
@article{rise2026,
title={RISE: Self-Improving Robot Policy with Compositional World Model},
author={Yang, Jiazhi and Lin, Kunyang and Li, Jinwei and Zhang, Wencong and Lin, Tianwei and Wu, Longyan and Su, Zhizhong and Zhao, Hao and Zhang, Ya-Qin and Chen, Li and Luo, Ping and Yue, Xiangyu and Li, Hongyang},
journal={arXiv preprint arXiv:2602.11075},
year={2026}
}