Skip to content

Commit f2810e5

Browse files
Add required_temp support in jsonl format to support flexible temperature setting for gen_api_answer (#2653)
1 parent 32c41de commit f2810e5

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

fastchat/llm_judge/gen_api_answer.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,13 @@
2727
def get_answer(
2828
question: dict, model: str, num_choices: int, max_tokens: int, answer_file: str
2929
):
30+
assert (
31+
args.force_temperature is not None and "required_temperature" in question.keys()
32+
) == False
3033
if args.force_temperature is not None:
3134
temperature = args.force_temperature
35+
elif "required_temperature" in question.keys():
36+
temperature = question["required_temperature"]
3237
elif question["category"] in temperature_config:
3338
temperature = temperature_config[question["category"]]
3439
else:

0 commit comments

Comments
 (0)