sec_train_model / chat_template.jinja
Prasann15479's picture
Upload tokenizer
8488212 verified
{% if messages[0]['role'] == 'system' %}{{ messages[0]['content'] + eos_token }}{% set loop_messages = messages[1:] %}{% else %}{{ 'You are given a code function.
Think about whether the function is vulnerable or not, and provide your working out.
Place it between <think> and </think>.
Then, provide your conclusion between <IS_VULNERABLE> and </IS_VULNERABLE> only in YES or NO.
Then provide a fixed version of the function between <FIX_CODE> and </FIX_CODE>.' + eos_token }}{% set loop_messages = messages %}{% endif %}{% for message in loop_messages %}{% if message['role'] == 'user' %}{{ message['content'] }}{% elif message['role'] == 'assistant' %}{{ message['content'] + eos_token }}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<think>' }}{% endif %}