Views
No views yet

1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4model_id = "tiiuae/Falcon3-7B-Base-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-7B-Base-1.58bit -q i2_s
python run_inference.py -m models/Falcon3-7B-1.58bit/ggml-model-i2_s.gguf -p "Hi how are you doing today?" -cnv| Benchmark | Llama3-8B-1.58-100B-tokens | Falcon3-7B-Base-1.58bit |
|---|---|---|
| IFEval | 17.91 | 25.43 |
| MUSR | 4.87 | 5.75 |
| GPQA | 1.83 | 2.32 |
| BBH | 5.36 | 3.91 |
| MMLU-PRO | 2.78 | 1.36 |
| MATH | 0.26 | 0.88 |
| Average | 5.5 | 6.61 |
@misc{Falcon3,
title = {The Falcon 3 Family of Open Models},
url = {https://huggingface.co/blog/falcon3},
author = {Falcon-LLM Team},
month = {December},
year = {2024}
}