Robotics
ONNX
Safetensors
Gr00tN1d7

update_onnx_model_prompt

#3

Updates the onnx models to use the newer prompt "move the apple to the plate" for consistency.

pulkitg-nv changed pull request status to open
NVIDIA org

Nit: can you fix the below change.
ReadMe line#20 The corrected language prompt, its 77-token vision-language sequence, embodiment ID, vision grid, normalization statistics, and static tensor shapes were captured by the trace. Re-export the policy when any of these profile values must change.

NVIDIA org

Is change in tensor shape expected? Is it a result of updating the language prompt?

Is change in tensor shape expected? Is it a result of updating the language prompt?

Yes, since the input tensor size changed from 90 something tokens to 77 the downstream graph connected to this input also changed.

Nit: can you fix the below change.
ReadMe line#20 The corrected language prompt, its 77-token vision-language sequence, embodiment ID, vision grid, normalization statistics, and static tensor shapes were captured by the trace. Re-export the policy when any of these profile values must change.

removed the line

pulkitg-nv changed pull request status to merged
pulkitg-nv deleted the refs/pr/3 ref

Sign up or log in to comment