Views
No views yet
1import numpy as np
2from huggingface_hub import snapshot_download
3from molmobot_pi0.eval.policies.pi import PiJointPosPolicy
4
5ckpt_dir = snapshot_download("allenai/MolmoBot-Pi0-DROID")
6policy = PiJointPosPolicy(checkpoint_dir=ckpt_dir)
7policy.prepare_model()
8
9obs = {
10 "task": "put the mug in the bowl",
11 "qpos": {
12 "arm": np.zeros(7),
13 "gripper": np.zeros(2),
14 },
15 "exo_camera_1": np.zeros((360, 640, 3), dtype=np.uint8),
16 "wrist_camera": np.zeros((360, 640, 3), dtype=np.uint8),
17}
18
19action = policy.get_action(obs)
20print(action)@misc{deshpande2026molmobot,
title={MolmoB0T: Large-Scale Simulation Enables Zero-Shot Manipulation},
author={Abhay Deshpande and Maya Guru and Rose Hendrix and Snehal Jauhri and Ainaz Eftekhar and Rohun Tripathi and Max Argus and Jordi Salvador and Haoquan Fang and Matthew Wallingford and Wilbert Pumacay and Yejin Kim and Quinn Pfeifer and Ying-Chun Lee and Piper Wolters and Omar Rayyan and Mingtong Zhang and Jiafei Duan and Karen Farley and Winson Han and Eli Vanderbilt and Dieter Fox and Ali Farhadi and Georgia Chalvatzaki and Dhruv Shah and Ranjay Krishna},
year={2026},
eprint={2603.16861},
archivePrefix={arXiv},
primaryClass={cs.RO},
url={https://arxiv.org/abs/2603.16861},
}