Views
No views yet

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