Views
No views yet
1def build_JET_chat_template(question, tokenizer):
2 system_prompt = (
3 "You are a helpful AI assistant. A conversation takes place between the User "
4 "and the Assistant. The User asks a question, and the Assistant solves it.\n"
5 "Please help me solve this question. Wrap only the final answer in \\boxed{}."
6 )
7 return tokenizer.apply_chat_template(
8 [
9 {"role": "system", "content": system_prompt},
10 {"role": "user", "content": question}
11 ],
12 tokenize=False,
13 add_generation_prompt=True
14 )