pipeline_tag: text-generation
Model Card for Breeze-7B-Base-v0.1
Breeze-7B-Base-v0.1 is a 7-billion-parameter language model built from Mistral-7B and tailored for Traditional Chinese (TC). This model expands the TC vocabulary (extra 30k TC tokens) based on the original Mistral-7B to better adapt to TC and improve inference speed, resulting in a doubling of the original tokenizer's inference speed. To the best of our knowledge, this is the first work on vocabulary expansion in TC. This model uses 250GB of TC data for continued pre-training. Breeze-7B-Base-v0.1 performs well on both EN and TC benchmarks. This model outperforms Taiwan-LLM-7B-v2.1-base, Taiwan-LLM-13B-v2.0-base, and Yi-6B-Base on all TC benchmarks and is comparable with Mistral-7B-v0.1 on MMLU and MT-Bench in English.
A project by the members (in alphabetical order): Chan-Jan Hsu 許湛然, Chang-Le Liu 劉昶樂, Feng-Ting Liao 廖峰挺, Po-Chun Hsu 許博竣, Yi-Chang Chen 陳宜昌, and the supervisor Da-Shan Shiu 許大山.
Features
- Expanding the vocabulary dictionary for Traditional Chinese from 32k to 62k vocabulary size
- 8k context length
Model Details
- Finetuned from: mistralai/Mistral-7B-v0.1
- Model type: Causal decoder-only transformer language model
- Language: English and Traditional Chinese (zh-tw)
Base Model Performance
| Models | TMMLU+ (ACC) | DRCD (EM) | MMLU (ACC) |
|---|---|---|---|
| 5 shot | 3 shot | 5 shot | |
| MediaTek-Research/Breeze-7B-Base-v0.1 | |||
| mistralai/Mistral-7B-v0.1 | |||
| yentinglin/Taiwan-LLM-7B-v2.1-base | |||
| yentinglin/Taiwan-LLM-13B-v2.0-base | |||
| 01-ai/Yi-6B | |||
| 01-ai/Yi-34B | |||
| Qwen/Qwen-7B | |||
| Qwen/Qwen-14B |
Inference Performance
| Models | Speed (char/sec) | Compression Ratio | Max Character Size |
|---|---|---|---|
| MediaTek-Research/Breeze-7B-Base-v0.1 | |||
| mistralai/Mistral-7B-v0.1 | |||
| yentinglin/Taiwan-LLM-7B-v2.1-base | |||
| yentinglin/Taiwan-LLM-13B-v2.0-base | |||
| 01-ai/Yi-6B | |||
| 01-ai/Yi-34B | |||
| Qwen/Qwen-7B | |||
| Qwen/Qwen-14B |
Chat Model Performance
| Models | TMMLU+ (ACC) | DRCD (EM) | MT-Bench-tw (Score) | MMLU (ACC) | MT-Bench (Score) |
|---|---|---|---|---|---|
| 5 shot | 3 shot | 0 shot | 5 shot | 0 shot | |
| MediaTek-Research/Breeze-7B-Instruct-v0.1 | |||||
| mistralai/Mistral-7B-Instruct-v0.1 | |||||
| yentinglin/Taiwan-LLM-7B-v2.1-chat | |||||
| yentinglin/Taiwan-LLM-13B-v2.0-chat | |||||
| 01-ai/Yi-6B-Chat | |||||
| 01-ai/Yi-34B-Chat | |||||
| Qwen/Qwen-7B-Chat | |||||
| Qwen/Qwen-14B-Chat | |||||
| gpt-3.5-turbo-0613 | 76.30 |
Use in Transformers
First install direct dependencies:
pip install transformers torch accelerate
If you want faster inference using flash-attention2, you need to install these dependencies:
pip install packaging ninja
pip install flash-attn
Then load the model in transformers:
from transformers import AutoModelForCausalLM, AutoTokenizer
import torch
model = AutoModelForCausalLM.from_pretrained(
model="MediaTek-Research/Breeze-7B-Base-v0.1",
device_map="auto",
torch_dtype=torch.bfloat16,
use_flash_attn_2=True # optional
)