| model_type: "BartChatbot" | |
| max_steps: 128 | |
| min_length: 2 | |
| repetition_penalty: 1 | |
| template_string: > | |
| _setting_name {{ location_name }} | |
| _setting_desc {{ location }}</s> | |
| _setting_name {{ name }} | |
| _setting_desc {{ persona }}</s> | |
| _setting_name {{ other_name }} | |
| _setting_desc {{ other_persona }}</s> | |
| {% for line in history -%} | |
| {% if line.speaker == name -%} | |
| <speaker_self> | |
| {%- elif line.speaker == other_name -%} | |
| <speaker_other> | |
| {%- endif -%} | |
| {{ line.line }} | |
| {%- if loop.index != (history|length) -%} | |
| </s> | |
| {% endif %} | |
| {%- 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 | |