Skip to content

Commit df9508f

Browse files
hjh0119Jintao-Huang
authored andcommitted
[bugfix] add tools fileds in grpo inputs2reqeusts (#6054)
1 parent 830f622 commit df9508f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

swift/trainers/rlhf_trainer/grpo_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2513,7 +2513,7 @@ def _process_image_data(image_data: Union[dict, str]) -> str:
25132513
return []
25142514

25152515
# Define core metadata fields required for all requests
2516-
REQUEST_METADATA_FIELDS = ['messages', 'images', 'audios', 'videos', 'objects', 'uuid']
2516+
REQUEST_METADATA_FIELDS = ['messages', 'images', 'audios', 'videos', 'tools', 'objects', 'uuid']
25172517
requests_dicts = []
25182518

25192519
for data in inputs:

0 commit comments

Comments
 (0)