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