Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model_id = "natong19/Mistral-Nemo-Instruct-2407-abliterated"
5device = "cuda"
6
7tokenizer = AutoTokenizer.from_pretrained(model_id)
8
9conversation = [{"role": "user", "content": "Where's the capital of France?"}]
10
11tool_use_prompt = tokenizer.apply_chat_template(
12 conversation,
13 tokenize=False,
14 add_generation_prompt=True,
15)
16
17inputs = tokenizer(tool_use_prompt, return_tensors="pt", return_token_type_ids=False).to(device)
18
19model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto")
20
21outputs = model.generate(**inputs, max_new_tokens=128)
22print(tokenizer.decode(outputs[0][len(inputs["input_ids"][0]):], skip_special_tokens=True))| Benchmark | Mistral-Nemo-Instruct-2407 | Mistral-Nemo-Instruct-2407-abliterated |
|---|---|---|
| ARC (25-shot) | 65.9 | 65.8 |
| GSM8K (5-shot) | 76.2 | 75.2 |
| HellaSwag (10-shot) | 84.3 | 84.3 |
| MMLU (5-shot) | 68.4 | 68.8 |
| TruthfulQA (0-shot) | 54.9 | 55.0 |
| Winogrande (5-shot) | 82.2 | 82.6 |