Views
No views yet

1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3
4model_name = "tiiuae/Falcon3-1B-Instruct"
5
6model = AutoModelForCausalLM.from_pretrained(
7 model_name,
8 torch_dtype="auto",
9 device_map="auto"
10)
11tokenizer = AutoTokenizer.from_pretrained(model_name)
12
13prompt = "How many hours in one day?"
14messages = [
15 {"role": "system", "content": "You are a helpful friendly assistant Falcon3 from TII, try to follow instructions as much as possible."},
16 {"role": "user", "content": prompt}
17]
18text = tokenizer.apply_chat_template(
19 messages,
20 tokenize=False,
21 add_generation_prompt=True
22)
23model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
24
25generated_ids = model.generate(
26 **model_inputs,
27 max_new_tokens=1024
28)
29generated_ids = [
30 output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
31]
32
33response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
34print(response)| Category | Benchmark | Llama-3.2-1B | Qwen2.5-1.5B | SmolLM2-1.7B | Falcon3-1B-Instruct |
|---|---|---|---|---|---|
| General | MMLU (5-shot) | 68.2 | 59.8 | 49.2 | 46.1 |
| MMLU-PRO (5-shot) | 16 | 28.2 | 20 | 18.6 | |
| IFEval | 55.3 | 44.2 | 53 | 54.4 | |
| Math | GSM8K (5-shot) | 82.6 | 57.8 | 47.6 | 43.9 |
| GSM8K (8-shot, COT) | 46.6 | 58.8 | 46.3 | 45.8 | |
| MATH Lvl-5 (4-shot) | 5.2 | 1.1 | 3.1 | 1 | |
| Reasoning | Arc Challenge (25-shot) | 58.6 | 50.7 | 49.7 | 47.7 |
| GPQA (0-shot) | 24.4 | 29.6 | 28.6 | 26.5 | |
| GPQA (0-shot, COT) | 13.2 | 9.2 | 16 | 21.3 | |
| MUSR (0-shot) | 32 | 36.5 | 32.9 | 40.7 | |
| BBH (3-shot) | 33.8 | 39.2 | 34 | 35.1 | |
| CommonSense Understanding | PIQA (0-shot) | 72.1 | 73.2 | 74.4 | 72 |
| SciQ (0-shot) | 61.8 | 69.5 | 71.4 | 86.8 | |
| Winogrande (0-shot) | - | - | - | 60.2 | |
| OpenbookQA (0-shot) | 40.2 | 40.4 | 42.8 | 40 | |
| MT-Bench (avg) | 5.4 | 7.1 | 6.1 | 5.5 | |
| Instructions following | Alpaca (WC) | 8.6 | 8.6 | 5.4 | 6.1 |
@misc{Falcon3,
title = {The Falcon 3 Family of Open Models},
url = {https://huggingface.co/blog/falcon3},
author = {Falcon-LLM Team},
month = {December},
year = {2024}
}