Views
No views yet
1import torch
2from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
3
4model_name = "xinchen9/Llama3.1_8B_Instruct_CoT"
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
7model.generation_config = GenerationConfig.from_pretrained(model_name)
8model.generation_config.pad_token_id = model.generation_config.eos_token_id| Metric | Value |
|---|---|
| Avg. | 15.73 |
| IFEval (0-Shot) | 30.03 |
| BBH (3-Shot) | 22.06 |
| MATH Lvl 5 (4-Shot) | 4.61 |
| GPQA (0-shot) | 7.16 |
| MuSR (0-shot) | 8.46 |
| MMLU-PRO (5-shot) | 22.04 |