1>>> from transformers import AutoModelForCausalLM, AutoTokenizer
2>>> tokenizer = AutoTokenizer.from_pretrained("MediaTek-Research/Breeze-7B-32k-Instruct-v1_0/")
3>>> model = AutoModelForCausalLM.from_pretrained(
4>>> "MediaTek-Research/Breeze-7B-32k-Instruct-v1_0",
5... device_map="auto",
6... torch_dtype=torch.bfloat16,
7... attn_implementation="flash_attention_2"
8... )
9>>> chat = [
10... {"role": "user", "content": "你好,請問你可以完成什麼任務?"},
11... {"role": "assistant", "content": "你好,我可以幫助您解決各種問題、提供資訊和協助您完成許多不同的任務。例如:回答技術問題、提供建議、翻譯文字、尋找資料或協助您安排行程等。請告訴我如何能幫助您。"},
12... {"role": "user", "content": "太棒了!"},
13... ]
14>>> tokenizer.apply_chat_template(chat, tokenize=False)
15"<s>You are a helpful AI assistant built by MediaTek Research. The user you are helping speaks Traditional Chinese and comes from Taiwan. [INST] 你好,請問你可以完成什麼任務? [/INST] 你好,我可以幫助您解決各種問題、提供資訊和協助您完成許多不同的任務。例如:回答技術問題、提供建議、翻譯文字、尋找資料或協助您安排行程等。請告訴我如何能幫助您。 [INST] 太棒了! [/INST] "
16# Tokenized results
17# ['▁', '你好', ',', '請問', '你', '可以', '完成', '什麼', '任務', '?']
18# ['▁', '你好', ',', '我', '可以', '幫助', '您', '解決', '各種', '問題', '、', '提供', '資訊', '和', '協助', '您', '完成', '許多', '不同', '的', '任務', '。', '例如', ':', '回答', '技術', '問題', '、', '提供', '建議', '、', '翻譯', '文字', '、', '尋找', '資料', '或', '協助', '您', '安排', '行程', '等', '。', '請', '告訴', '我', '如何', '能', '幫助', '您', '。']
19# ['▁', '太', '棒', '了', '!']