molmoact2-space / README.md
MichaelMintIcecream's picture
spike: MolmoAct2 Docker Space (server + Dockerfile)
40fa6ec verified
|
Raw
History Blame Contribute Delete
1.39 kB
metadata
title: MolmoAct2 SO100 101 Inference
emoji: πŸ€–
colorFrom: indigo
colorTo: blue
sdk: docker
app_port: 7860
pinned: false
short_description: Nori cloud-inference server for MolmoAct2-SO100_101 (/act)

MolmoAct2-SO100_101 β€” Nori cloud inference

Private Docker Space serving allenai/MolmoAct2-SO100_101 for Nori robot rollout. It runs our own FastAPI/uvicorn server (molmoact2_server.py) β€” not the HF Inference-Endpoint toolkit, which is incompatible with the transformers version this model needs.

Endpoints

  • GET /health β†’ {"ok", "status": "loading|ready|error", "error", "repo", "dtype"}
  • POST /act (Bearer NORI_INFER_TOKEN) β†’ { images:[b64...], state:[6 floats], instruction:str, num_steps? } β†’ { actions: [[...6 DOF...], ... up to 30 moves] } (robot scale).

Required setup (Space Settings)

  1. Hardware: a GPU tier β€” a10g-small (A10G 24GB, ~$1/hr) is enough (bf16 <16GB).
  2. Secrets:
    • NORI_INFER_TOKEN β€” the bearer token the rollout client sends (required).
    • HF_TOKEN β€” only if the model repo is gated (allenai's is public; usually not needed).
  3. First boot downloads ~21GB, so /health reports "loading" for a few minutes, then "ready". Add persistent storage later to skip re-downloads on restart.

Deploy/update instructions: see cloud_inference/space/DEPLOY.md in the Nori-Lab repo.