mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-23 12:40:08 +00:00
fix(rewards/robometer): add missing input keys mm_token_type_ids
This commit is contained in:
@@ -78,6 +78,7 @@ ROBOMETER_QWEN_INPUT_KEYS = (
|
||||
"image_grid_thw",
|
||||
"video_grid_thw",
|
||||
"second_per_grid_ts",
|
||||
"mm_token_type_ids",
|
||||
)
|
||||
ROBOMETER_METADATA_KEYS = (
|
||||
"prog_token_id",
|
||||
|
||||
Reference in New Issue
Block a user