From c1fdf81df6ade5da7be4eb66b715f0efd171d5aa Mon Sep 17 00:00:00 2001
From: hiyouga <467089858@qq.com>
Date: Mon, 27 May 2024 20:54:26 +0800
Subject: [PATCH] tiny fix

---
 src/llamafactory/data/template.py    | 3 +--
 src/llamafactory/extras/constants.py | 1 +
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/llamafactory/data/template.py b/src/llamafactory/data/template.py
index 5a12524d..8332b0cb 100644
--- a/src/llamafactory/data/template.py
+++ b/src/llamafactory/data/template.py
@@ -742,7 +742,6 @@ _register_template(
 _register_template(
     name="olmo",
     format_user=StringFormatter(slots=["<|user|>\n{{content}}<|assistant|>\n"]),
-    format_assistant=StringFormatter(slots=["{{content}}", {"eos_token"}]),
     format_system=StringFormatter(slots=[{"eos_token"}, "{{content}}"]),
     force_system=True,
 )
@@ -751,11 +750,11 @@ _register_template(
 _register_template(
     name="openchat",
     format_user=StringFormatter(slots=["GPT4 Correct User: {{content}}", {"eos_token"}, "GPT4 Correct Assistant:"]),
-    format_assistant=StringFormatter(slots=["{{content}}", {"eos_token"}]),
     format_system=StringFormatter(slots=[{"bos_token"}, "{{content}}"]),
     force_system=True,
 )
 
+
 _register_template(
     name="openchat-3.6",
     format_user=StringFormatter(
diff --git a/src/llamafactory/extras/constants.py b/src/llamafactory/extras/constants.py
index 3cb221bf..d1fcd4f6 100644
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -719,6 +719,7 @@ register_model_group(
     template="openchat",
 )
 
+
 register_model_group(
     models={
         "OpenChat3.6-8B-Chat": {