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