baasfasfasfasf's picture
Correct prompt template
fdb5b37
raw
history blame
846 Bytes
model_type: "BartChatbot"
max_steps: 128
min_length: 2
repetition_penalty: 1
template_string: >
_setting_name {{ location_name }}
_setting_desc {{ location }}</s>
_self_name {{ name }}
_self_persona {{ persona }}</s>
_partner_name {{ other_name }}
_other_persona {{ other_persona }}</s>
{% for line in history -%}
{% if line.speaker == name -%}
<speaker_self>
{%- elif line.speaker == other_name -%}
<speaker_other>
{%- endif -%}
{{- line.line -}}
</s>
{% endfor -%}
default_context: >
{
"location_name": "",
"location": "",
"name": "",
"persona": "",
"other_name": "",
"other_persona": "",
"history": [
{"speaker": "", "line": ""},
{"speaker": "", "line": ""}
]
}
bos_token_id: 0
eos_token_id: 2
pad_token_id: 1