-
Notifications
You must be signed in to change notification settings - Fork 118
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add mt-bench #75
Add mt-bench #75
Conversation
@NathanHB feel free to ping me once it's merged with main so we can integrate it :) |
Careful with the deletion of task_examples, quite sure some of these files are needed by the nanotron team. |
src/lighteval/models/base_model.py
Outdated
@@ -345,6 +353,91 @@ def greedy_until_with_logits( | |||
override_bs=override_bs, | |||
) | |||
|
|||
def greedy_until_multi_turn(self, requests: list[GreedyUntilMultiTurnRequest], override_bs: Optional[int] = None) -> GenerateMultiTurnReturn: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
You'll also need to update the other model launchers, or at least add a function in the abstract model which indicates that it needs to be implemented
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Quite cool overall!
Some code reorgs are need imo (management of multi turn context with all the other contexts, finding a better way to pass judgements for logging, implementing greedy_until_multi_turn
in the other models, or at least the LightevalModel
class), plus some doc.
Nice that the code is so neat, it looks like it will be easy to switch to "using a judge as metric" in the future!
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM, minor questions.
Overall, very good job in adding this mechanism! Looking forward having it as an actual metric!
What this PR does:
This implementation uses mt-bench prompts from InflectionAI. The code is inspired from the original implementation of mt-bench with notable differences.
do_sample=False
and temperature set to0.0
.