Views
No views yet

1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4model_id = "tiiuae/Falcon3-7B-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-10B-Instruct-1.58bit -q i2_s
python run_inference.py -m models/Falcon3-10B-1.58bit/ggml-model-i2_s.gguf -p "You are a helpful assistant" -cnv| Benchmark | Llama3-8B-1.58-100B-tokens | Falcon3-10B-Instruct-1.58bit |
|---|---|---|
| IFEval | 17.91 | 54.37 |
| MUSR | 4.87 | 2.57 |
| GPQA | 1.83 | 4.27 |
| BBH | 5.36 | 6.59 |
| MMLU-PRO | 2.78 | 6.62 |
| MATH | 0.26 | 2.44 |
| Average | 5.5 | 12.81 |
@misc{Falcon3,
title = {The Falcon 3 Family of Open Models},
author = {Falcon-LLM Team},
month = {December},
year = {2024}
}