How to use from the
Use from the
LeRobot library
# No code snippets available yet for this library.

# To use this model, check the repository files and the library's documentation.

# Want to help? PRs adding snippets are welcome at:
# https://github.com/huggingface/huggingface.js

molmoact2_tool_insert

Fine-tuned MolmoAct2 (action-expert-only) on SO101 data.

Base model allenai/MolmoAct2
Dataset villekuosmanen/armnetbench_tool_insert
Task tool_insert
Training Vast.ai H100 SXM, batch 64, bf16, gradient checkpointing
W&B project molmoact2_tool_insert
W&B run 8m94f03c

Checkpoints

Step Path
006000 checkpoints/006000/pretrained_model/
004000 checkpoints/004000/pretrained_model/

Usage

from lerobot.policies.molmoact2.modeling_molmoact2 import MolmoAct2Policy
policy = MolmoAct2Policy.from_pretrained("pravsels/molmoact2_tool_insert", revision="main")
Downloads last month
9
Safetensors
Model size
5B params
Tensor type
F32
·
BF16
·
Video Preview
loading

Model tree for pravsels/molmoact2_tool_insert

Finetuned
(53)
this model