Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "newsbang/Homer-v1.0-Qwen2.5-7B"
4
5model = AutoModelForCausalLM.from_pretrained(
6 model_name,
7 torch_dtype="auto",
8 device_map="auto"
9)
10tokenizer = AutoTokenizer.from_pretrained(model_name)
11
12messages = [
13 {"role": "system", "content": "You are a very helpful assistant."},
14 {"role": "user", "content": "Hello"}
15]
16text = tokenizer.apply_chat_template(
17 messages,
18 tokenize=False,
19 add_generation_prompt=True
20)
21model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
22
23generated_ids = model.generate(
24 **model_inputs,
25 max_new_tokens=512
26)
27generated_ids = [
28 output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
29]
30
31response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]| Metric | Value |
|---|---|
| Avg. | 32.15 |
| IFEval (0-Shot) | 63.93 |
| BBH (3-Shot) | 37.81 |
| MATH Lvl 5 (4-Shot) | 30.36 |
| GPQA (0-shot) | 9.62 |
| MuSR (0-shot) | 11.88 |
| MMLU-PRO (5-shot) | 39.27 |