add Exaone3.0 template

Former-commit-id: f7478af1d04353ab13236323e3bfb96fd2870fce
2024-09-30 09:17:41 +09:00
parent e702678252
commit f0f80819a0
2 changed files with 21 additions and 0 deletions
--- a/src/llamafactory/data/template.py
+++ b/src/llamafactory/data/template.py
@@ -644,6 +644,18 @@ _register_template(
 )


+_register_template(
+    name="exaone",
+    format_user=StringFormatter(slots=["[|user|]{{content}}\n[|assistant|]"]),
+    format_system=StringFormatter(slots=["[|system|]{{content}}[|endofturn|]\n"]),
+    format_assistant=StringFormatter(slots=["{{content}}[|endofturn|]\n"]),
+    format_separator=EmptyFormatter(slots=["\n"]),
+    format_prefix=EmptyFormatter(slots=[""]),
+    stop_words=["[|endofturn|]"],
+    efficient_eos=True,
+)
+
+
 _register_template(
    name="falcon",
    format_user=StringFormatter(slots=["User: {{content}}\nFalcon:"]),