From f2533a2800483d7272dbc5ee07c9242e6e7cdecb Mon Sep 17 00:00:00 2001 From: YC Chen Date: Wed, 2 Aug 2023 14:35:35 +0800 Subject: [PATCH] [fix] Remove useless code Former-commit-id: 077e1556112913e4eeef47e581055183b39d5404 --- src/llmtuner/extras/template.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/llmtuner/extras/template.py b/src/llmtuner/extras/template.py index e85e0f35..567b9102 100644 --- a/src/llmtuner/extras/template.py +++ b/src/llmtuner/extras/template.py @@ -65,7 +65,7 @@ class Llama2Template(Template): if turn_idx == 0: convs.append([self.prompt.format(query=sys+query_i), resp_i]) else: - convs.append([(self.sep if turn_idx else prefix) + self.prompt.format(query=query_i), resp_i]) + convs.append([self.sep + self.prompt.format(query=query_i), resp_i]) return convs