Falcon3-1B-Base / README.md
ybelkada's picture
Update README.md (#9)
bb1214f verified
|
raw
history blame
6 kB
metadata
language:
  - en
  - fr
  - es
  - pt
tags:
  - falcon3
license: other
license_name: falcon-llm-license
license_link: https://falconllm.tii.ae/falcon-terms-and-conditions.html
library_name: transformers
drawing

Falcon3-1B-Base

Falcon3 family of Open Foundation Models is a set of pretrained and instruct LLMs ranging from 1B to 10B parameters.

This repository contains the Falcon3-1B-Base. It achieves strong results on reasoning, language understanding, instruction following, code and mathematics tasks. Falcon3-1B-Base supports 4 languages (English, French, Spanish, Portuguese) and a context length of up to 4K. It was pruned in terms of depth, width, number of heads, and embedding channels from a larger 3B Falcon model, and was efficiently trained on only 80 GT using a knowledge distillation objective.

⚠️ This is a raw, pretrained model, which should be further finetuned using SFT, RLHF, continued pretraining, etc. for most use cases.

Model Details

  • Architecture
    • Transformer-based causal decoder-only architecture
    • 18 decoder blocks
    • Grouped Query Attention (GQA) for faster inference: 8 query heads and 4 key-value heads
    • Wider head dimension: 256
    • High RoPE value to support long context understanding: 1000042
    • Uses SwiGLU and RMSNorm
    • 4K context length
    • 131K vocab size
  • Pruned and healed using larger Falcon models (3B and 7B respectively) on only 80 Gigatokens of datasets comprising of web, code, STEM, high quality and multilingual data using 256 H100 GPU chips
  • Supports EN, FR, ES, PT
  • Developed by Technology Innovation Institute
  • License: TII Falcon-LLM License 2.0
  • Model Release Date: December 2024

Getting started

Click to expand
import torch
from transformers import pipeline

pipe = pipeline(
    "text-generation", 
    model="tiiuae/Falcon3-1B-Base", 
    torch_dtype=torch.bfloat16, 
    device_map="auto"
)
response = pipe("Question: How many hours in one day? Answer: ")
print(response[0]['generated_text'])

Benchmarks

We report in the following table our internal pipeline benchmarks:

Category Benchmark Llama-3.2-1B Qwen2.5-1.5B SmolLM2-1.7B Falcon3-1B-Base
General MMLU (5-shot) 31.1 61.0 50.1 42.5
MMLU-PRO (5-shot) 11.7 28.4 21.3 16.1
IFEval 14.8 26.0 24.2 25.2
Math GSM8K (5-shot) 6.6 62.2 31.0 34.3
MATH Lvl-5 (4-shot) 0.2 6.7 1.4 2.2
Reasoning Arc Challenge (25-shot) 40.2 54.8 54.1 48.1
GPQA (0-shot) 24.2 28.1 28.9 28.1
MUSR (0-shot) 34.5 35.5 34.7 41.9
BBH (3-shot) 31.2 41.1 34.2 36.0
CommonSense Understanding PIQA (0-shot) 74.5 76.0 77.5 74.5
SciQ (0-shot) 88.5 93.1 90.8 91.1
Winogrande (0-shot) 60.4 63.0 66.1 61.2
OpenbookQA (0-shot) 37.4 40.4 44.0 41.0

Useful links

Technical Report

Coming soon....

Citation

If the Falcon3 family of models were helpful to your work, feel free to give us a cite.

@misc{Falcon3,
    title = {The Falcon 3 Family of Open Models},
    url = {https://huggingface.co/blog/falcon3},
    author = {Falcon-LLM Team},
    month = {December},
    year = {2024}
}