Views
No views yet

1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3model_name = "tiiuae/Falcon3-3B-Instruct"
4
5model = AutoModelForCausalLM.from_pretrained(
6 model_name,
7 torch_dtype="auto",
8 device_map="auto"
9)
10tokenizer = AutoTokenizer.from_pretrained(model_name)
11
12prompt = "How many hours in one day?"
13messages = [
14 {"role": "system", "content": "You are a helpful friendly assistant Falcon3 from TII, try to follow instructions as much as possible."},
15 {"role": "user", "content": prompt}
16]
17text = tokenizer.apply_chat_template(
18 messages,
19 tokenize=False,
20 add_generation_prompt=True
21)
22model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
23
24generated_ids = model.generate(
25 **model_inputs,
26 max_new_tokens=1024
27)
28generated_ids = [
29 output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
30]
31
32response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
33print(response)| Category | Benchmark | Llama-3.2-3B-Instruct | Qwen2.5-3B-Instruct | Nemotron-Mini-4B-Instruct | Falcon3-3B-Instruct |
|---|---|---|---|---|---|
| General | MMLU (5-shot) | 61.2 | 65.4 | 57.3 | 56.9 |
| MMLU-PRO (5-shot) | 27.7 | 32.6 | 26.0 | 29.7 | |
| IFEval | 74.7 | 64.1 | 66.3 | 68.3 | |
| Math | GSM8K (5-shot) | 76.8 | 56.7 | 29.8 | 74.8 |
| GSM8K (8-shot, COT) | 78.8 | 60.8 | 35.0 | 78.0 | |
| MATH Lvl-5 (4-shot) | 14.6 | 0.0 | 0.0 | 19.9 | |
| Reasoning | Arc Challenge (25-shot) | 50.9 | 55.0 | 56.2 | 55.5 |
| GPQA (0-shot) | 32.2 | 29.2 | 27.0 | 29.6 | |
| GPQA (0-shot, COT) | 11.3 | 11.0 | 12.2 | 26.5 | |
| MUSR (0-shot) | 35.0 | 40.2 | 38.7 | 39.0 | |
| BBH (3-shot) | 41.8 | 44.5 | 39.5 | 45.4 | |
| CommonSense Understanding | PIQA (0-shot) | 74.6 | 73.8 | 74.6 | 75.6 |
| SciQ (0-shot) | 77.2 | 60.7 | 71.0 | 95.5 | |
| Winogrande (0-shot) | - | - | - | 65.0 | |
| OpenbookQA (0-shot) | 40.8 | 41.2 | 43.2 | 42.2 | |
| Instructions following | MT-Bench (avg) | 7.1 | 8.0 | 6.7 | 7.2 |
| Alpaca (WC) | 19.4 | 19.4 | 9.6 | 15.5 | |
| Tool use | BFCL AST (avg) | 85.2 | 84.8 | 59.8 | 59.3 |
| Code | EvalPlus (0-shot) (avg) | 55.2 | 69.4 | 40.0 | 52.9 |
| Multipl-E (0-shot) (avg) | 31.6 | 29.2 | 19.6 | 32.9 |
@misc{Falcon3,
title = {The Falcon 3 Family of Open Models},
url = {https://huggingface.co/blog/falcon3},
author = {Falcon-LLM Team},
month = {December},
year = {2024}
}