From ed14f8bae753872cf9744d592f070265b25823b6 Mon Sep 17 00:00:00 2001 From: codingma Date: Sun, 7 Apr 2024 11:39:54 +0800 Subject: [PATCH] rename template to breeze Former-commit-id: 5a780e9eec7e8c560d417c1a95afa1be0f878d32 --- src/llmtuner/data/template.py | 2 +- src/llmtuner/extras/constants.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/llmtuner/data/template.py b/src/llmtuner/data/template.py index 440030db..bd9e65e9 100644 --- a/src/llmtuner/data/template.py +++ b/src/llmtuner/data/template.py @@ -775,7 +775,7 @@ _register_template( ) _register_template( - name="mediatek", + name="breeze", format_user=StringFormatter(slots=["[INST] {{content}} [/INST] "]), format_system=StringFormatter(slots=[{"bos_token"}, "{{content}}"]), default_system="You are a helpful AI assistant built by MediaTek Research. The user you are helping speaks Traditional Chinese and comes from Taiwan. ", diff --git a/src/llmtuner/extras/constants.py b/src/llmtuner/extras/constants.py index 535c7cdb..b2ee3058 100644 --- a/src/llmtuner/extras/constants.py +++ b/src/llmtuner/extras/constants.py @@ -981,5 +981,5 @@ register_model_group( DownloadSource.MODELSCOPE: "MediaTek-Research/Breeze-7B-Instruct-v1_0", } }, - template="mediatek", + template="breeze", ) \ No newline at end of file