Views
No views yet
TigerResearch/pretrain_zh dataset, a comprehensive Chinese pre-training dataset provided by TigerResearch. For more information about the dataset, please visit: TigerResearch/pretrain_zh.| Model | ceval | cmmlu | mmlu |
|---|---|---|---|
| Qwen1.5-7B | 73.85 | 73.29 | 60.31 |
| Qwen1.5-7B-filter | 64.34 | 64.48 | 60.49 |
| Qwen1.5-7B-Character | 67.24 | 65.45 | 59.99 |
1from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
2
3model_name = 'Henry94/Qwen1.5-7B-Character'
4
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", device_map="auto")
7
8
9prompt = "请简单介绍一下大型语言模型."
10messages = [
11 {"role": "system", "content": "You are Qwen, created by Alibaba Cloud. You are a helpful assistant."},
12 {"role": "user", "content": prompt}
13]
14text = tokenizer.apply_chat_template(
15 messages,
16 tokenize=False,
17 add_generation_prompt=True
18)
19model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
20
21generated_ids = model.generate(
22 **model_inputs,
23 max_new_tokens=512
24)
25generated_ids = [
26 output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
27]
28
29response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
30
31print(response)