Views
No views yet

1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3
4from transformers import AutoModelForCausalLM, AutoTokenizer
5
6model_name = "tiiuae/Falcon3-7B-Instruct"
7
8model = AutoModelForCausalLM.from_pretrained(
9 model_name,
10 torch_dtype="auto",
11 device_map="auto"]
12)
13tokenizer = AutoTokenizer.from_pretrained(model_name)
14
15prompt = "How many hours in one day?"
16messages = [
17 {"role": "system", "content": "You are a helpful friendly assistant Falcon3 from TII, try to follow instructions as much as possible."},
18 {"role": "user", "content": prompt}
19]
20text = tokenizer.apply_chat_template(
21 messages,
22 tokenize=False,
23 add_generation_prompt=True
24)
25model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
26
27generated_ids = model.generate(
28 **model_inputs,
29 max_new_tokens=1024
30)
31generated_ids = [
32 output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
33]
34
35response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
36print(response)| Category | Benchmark | Llama-3.1-8B-Instruct | Qwen2.5-7B-Instruct | Falcon3-7B-Instruct |
|---|---|---|---|---|
| General | MMLU (5-shot) | 55.9 | 72.4 | 68 |
| MMLU-PRO (5-shot) | 21.8 | 35.8 | 40.7 | |
| IFEval | 78.8 | 74.7 | 76.5 | |
| Math | GSM8K (5-shot) | 78.1 | 77.5 | 79.1 |
| GSM8K (8-shot, COT) | 79.8 | 72.7 | 80.9 | |
| MATH Lvl-5 (4-shot) | 10.4 | 26 | 29.4 | |
| Reasoning | Arc Challenge (25-shot) | 46.6 | 55.7 | 65.9 |
| GPQA (0-shot) | 33.6 | 31.9 | 32 | |
| GPQA (0-shot, COT) | 9.6 | 13.8 | 22.3 | |
| MUSR (0-shot) | 38.6 | 40.7 | 46.4 | |
| BBH (3-shot) | 43.7 | 53.9 | 52.4 | |
| CommonSense Understanding | PIQA (0-shot) | 78.9 | 73.7 | 78.8 |
| SciQ (0-shot) | 80.2 | 50.9 | 94.7 | |
| Winogrande (0-shot) | - | - | 70.4 | |
| OpenbookQA (0-shot) | 46.2 | 42.4 | 45.8 | |
| Instructions following | MT-Bench (avg) | 7.9 | 8.5 | 8.4 |
| Alpaca (WC) | 26.6 | 31.5 | 26.1 | |
| Tool use | BFCL AST (avg) | 90.6 | 91.4 | 72.3 |
@misc{Falcon3,
title = {The Falcon 3 family of Open Models},
author = {TII Team},
month = {December},
year = {2024}
}