admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-27 14:39:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Khalil Meftah
2026-05-21 11:05:02 +02:00
parent f6ecb7b955
commit 5cfca59ec7
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/rewards/topreward/processor_topreward.py
+1
View File
@@ -65,6 +65,7 @@ TOPREWARD_VLM_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",
) )
TOPREWARD_METADATA_KEYS = ("prompt_length",) TOPREWARD_METADATA_KEYS = ("prompt_length",)
TOPREWARD_INPUT_KEYS = TOPREWARD_VLM_INPUT_KEYS + TOPREWARD_METADATA_KEYS TOPREWARD_INPUT_KEYS = TOPREWARD_VLM_INPUT_KEYS + TOPREWARD_METADATA_KEYS