Views
No views yet

1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4model_id = "tiiuae/Falcon3-3B-Instruct-1.58bit"
5
6model = AutoModelForCausalLM.from_pretrained(
7 model_id,
8 torch_dtype=torch.bfloat16,
9).to("cuda")
10
11# Perform text generationgit clone https://github.com/microsoft/BitNet && cd BitNet
pip install -r requirements.txt
python setup_env.py --hf-repo tiiuae/Falcon3-3B-Instruct-1.58bit -q i2_s
python run_inference.py -m models/Falcon3-3B-1.58bit/ggml-model-i2_s.gguf -p "You are a helpful assistant." -cnv| Benchmark | Llama3-8B-1.58-100B-tokens | Falcon3-3B-Instruct-1.58bit |
|---|---|---|
| IFEval | 17.91 | 32.52 |
| MUSR | 4.87 | 2.23 |
| GPQA | 6.95 | 5.25 |
| BBH | 5.36 | 5.79 |
| MMLU-PRO | 2.78 | 3.41 |
| MATH | 0.26 | 0.77 |
| Average | 5.5 | 8.61 |
@misc{Falcon3,
title = {The Falcon 3 Family of Open Models},
author = {Falcon-LLM Team},
month = {December},
year = {2024}
}