Build with the full FlashRT runtime.
import flash_rt
model = flash_rt.load_model(
checkpoint="/path/to/checkpoint",
framework="torch", # or "jax"
autotune=3, # 0=off, 3=default, 5=thorough
)
actions = model.predict(
images=[base_img, wrist_img],
prompt="pick up the red block",
)