baasfasfasfasf commited on
Commit
fdb5b37
1 Parent(s): 748ee83

Correct prompt template

Browse files
Files changed (1) hide show
  1. config.yml +16 -10
config.yml CHANGED
@@ -4,22 +4,28 @@ min_length: 2
4
  repetition_penalty: 1
5
  template_string: >
6
  _setting_name {{ location_name }}
 
7
  _setting_desc {{ location }}</s>
8
- _setting_name {{ name }}
9
- _setting_desc {{ persona }}</s>
10
- _setting_name {{ other_name }}
11
- _setting_desc {{ other_persona }}</s>
 
 
 
 
 
12
  {% for line in history -%}
13
  {% if line.speaker == name -%}
14
- <speaker_self>
15
  {%- elif line.speaker == other_name -%}
16
- <speaker_other>
17
  {%- endif -%}
18
- {{ line.line }}
19
- {%- if loop.index != (history|length) -%}
20
  </s>
21
- {% endif %}
22
- {%- endfor %}
 
23
  default_context: >
24
  {
25
  "location_name": "",
 
4
  repetition_penalty: 1
5
  template_string: >
6
  _setting_name {{ location_name }}
7
+
8
  _setting_desc {{ location }}</s>
9
+
10
+ _self_name {{ name }}
11
+
12
+ _self_persona {{ persona }}</s>
13
+
14
+ _partner_name {{ other_name }}
15
+
16
+ _other_persona {{ other_persona }}</s>
17
+
18
  {% for line in history -%}
19
  {% if line.speaker == name -%}
20
+ <speaker_self>
21
  {%- elif line.speaker == other_name -%}
22
+ <speaker_other>
23
  {%- endif -%}
24
+ {{- line.line -}}
 
25
  </s>
26
+
27
+ {% endfor -%}
28
+
29
  default_context: >
30
  {
31
  "location_name": "",