1234567891011121314151617 |
- {%- for message in messages -%}
- {%- if message['role'] == 'system' -%}
- {{- 'System: ' + message['content'] -}}
- {%- elif message['role'] == 'user' -%}
- {{- 'User: ' + message['content'] -}}
- {%- elif message['role'] == 'assistant' -%}
- {{- 'Falcon: ' + message['content'] -}}
- {%- endif -%}
- {%- if (loop.last and add_generation_prompt) or not loop.last -%}
- {{- '\n' -}}
- {%- endif -%}
- {%- endfor -%}
- {%- if add_generation_prompt and messages[-1]['role'] != 'assistant' -%}
- {{- 'Falcon:' -}}
- {% endif %}
|