admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-24 21:19:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(rewards/robometer): add missing input keys mm_token_type_ids

Browse Source
This commit is contained in:
Khalil Meftah
2026-05-21 11:15:49 +02:00
parent bcd1ace38d
commit 1b559e1c84
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/rewards/robometer/modeling_robometer.py
+1
View File
@@ -78,6 +78,7 @@ ROBOMETER_QWEN_INPUT_KEYS = (
"image_grid_thw", "image_grid_thw",
"video_grid_thw", "video_grid_thw",
"second_per_grid_ts", "second_per_grid_ts",
"mm_token_type_ids",
) )
ROBOMETER_METADATA_KEYS = ( ROBOMETER_METADATA_KEYS = (
"prog_token_id", "prog_token_id",