Initial commit

added_tokens.txt

@@ -0,0 +1 @@
+{"_other_persona": 50270, "<speaker_self>": 50272, "_self_persona": 50269, "_self_name": 50268, "<speaker_other>": 50271, "_partner_name": 50267, "_setting_desc": 50266, "_setting_name": 50265}

config.yml

@@ -0,0 +1,39 @@
+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

decoder_bart.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:193c8d322b2a09c09ab116ba9cfa3736b6a4ac939f5c47213933fb8aebfa8923
+size 1222627449

encoder_bart.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9d3a906776a689d69ce50117e894e3c45f1087e8dd64d8b48ea59072010944fe
+size 814818963