From 9f364d3880376f5a6c2ab490f465c0782a9dbd61 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Sun, 19 Nov 2023 16:29:30 +0800 Subject: [PATCH] fix Mistral template https://github.com/lm-sys/FastChat/pull/2547 Former-commit-id: d426ecdf6e95402fc36893f7e4f17f881e1b957b --- src/llmtuner/data/template.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/llmtuner/data/template.py b/src/llmtuner/data/template.py index 1bf24063..b852eb3d 100644 --- a/src/llmtuner/data/template.py +++ b/src/llmtuner/data/template.py @@ -499,7 +499,9 @@ register_template( "[INST] {{query}} [/INST]" ], system="", - sep=[] + sep=[ + " " + ] )