File size: 846 Bytes
748ee83 fdb5b37 748ee83 fdb5b37 748ee83 fdb5b37 748ee83 fdb5b37 748ee83 fdb5b37 748ee83 fdb5b37 748ee83 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
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
|