Update README.md

README.md

@@ -1,79 +1,137 @@
 ---
 language:
 - en
-- fr
 - es
 - pt
 tags:
 - falcon3
-license: other
-license_name: falcon-llm-license
+license: other 
+license_name: falcon-llm-license 
 license_link: https://falconllm.tii.ae/falcon-terms-and-conditions.html
-library_name: transformers
 ---
 
-<div align="center">
-    <img src="https://huggingface.co/datasets/tiiuae/documentation-images/resolve/main/general/falco3-logo.png" alt="drawing" width="500"/>
-</div>
 
-# Falcon3-1B-Base
 
-**Falcon3** family of Open Foundation Models is a set of pretrained and instruct LLMs ranging from 1B to 10B parameters.
+#  Table of Contents
 
-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.
+0. [TL;DR](#TL;DR)
+1. [Model Details](#model-details)
+2. [Usage](#usage)
+3. [Training Details](#training-details)
+4. [Evaluation](#evaluation)
 
-⚠️ **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](https://www.tii.ae)
-- License: TII Falcon-LLM License 2.0
-- Model Release Date: December 2024
+# TL;DR
 
+# Model Details
 
-## Getting started
+⚠️ **This is a raw, pretrained model, which should be further finetuned for most usecases.** 
+
+## Model Description
+
+- **Developed by:** [https://www.tii.ae](https://www.tii.ae)
+- **Model type:** Causal decoder-only
+- **Architecture:** Transformer-base
+- **Language(s) (NLP):** Mainly English
+- **License:** TII Falcon-LLM License 2.0
+
+<br>
+
+# Usage
+
+Find below some example scripts on how to use the model in `transformers` (Make sure to have the latest transformers, or the one built from source):
+
+## Using the Pytorch model with 🤗 transformers
+
+### Running the model on a CPU
+
+<details>
+<summary> Click to expand </summary>
+
+```python
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("tiiuae/Falcon3-7B-Base")
+model = AutoModelForCausalLM.from_pretrained("tiiuae/Falcon3-7B-Base")
+
+input_text = "Question: How many hours in one day? Answer: "
+input_ids = tokenizer(input_text, return_tensors="pt").input_ids
+
+outputs = model.generate(input_ids)
+print(tokenizer.decode(outputs[0]))
+```
+
+</details>
+
+### Running the model on a GPU
+
+<details>
+<summary> Click to expand </summary>
+
+```python
+# pip install accelerate
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("tiiuae/Falcon3-7B-Base")
+model = AutoModelForCausalLM.from_pretrained("tiiuae/Falcon3-7B-Base", device_map="auto")
+
+input_text = "Question: How many hours in one day? Answer: "
+input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to("cuda")
+
+outputs = model.generate(input_ids)
+print(tokenizer.decode(outputs[0]))
+```
+
+</details>
+
+### Running the model on a GPU using `torch.compile`
 
 <details>
 <summary> Click to expand </summary>
 
 ```python
 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'])
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("tiiuae/Falcon3-7B-Base")
+model = AutoModelForCausalLM.from_pretrained("tiiuae/Falcon3-7B-Base", torch_dtype=torch.bfloat16).to(0)
+
+model = torch.compile(model)
+
+input_text = "Question: How many hours in one day? Answer: "
+input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to("cuda")
+
+outputs = model.generate(input_ids)
+print(tokenizer.decode(outputs[0]))
 ```
 
 </details>
 
-<br>
 
-## Benchmarks
-We report in the following table our internal pipeline benchmarks.
- - We use [lm-evaluation harness](https://github.com/EleutherAI/lm-evaluation-harness).
- - We report **raw scores**.
- - We use same batch-size across all models.
+# Training Details
 
+## Training Data
 
+Falcon3-7B is trained on 15 Gigatokens of datasets comprising of web, code, STEM, high quality and mutlilingual data.
 
+## Training Procedure
+
+Falcon3-7B is trained on 256 H100 nodes (world size 2048).
+
+### Training Hyperparameters
+
+| **Hyperparameter** | **Value**  | **Comment**                           |
+|--------------------|------------|---------------------------------------|
+| Precision          | `bfloat16` |                                       |
+| Optimizer          | AdamW      |                                       |
+| Max learning rate  | 6e-4       | Following a WSD (warmup-stable-decay) |
+|                    |            | learning rate scheduler               |
+| Weight decay       | 1e-1       |                                       |
+| z-loss             | 1e-4       |                                       |
+| Batch size         | Variable   | Batch size was gradually increased    |
+|                    |            | during the training                   |
+
+# Evaluation
 <table border="1" style="width: 100%; text-align: center; border-collapse: collapse;">
     <colgroup>
         <col style="width: 10%;">
@@ -81,6 +139,7 @@ We report in the following table our internal pipeline benchmarks.
         <col style="width: 7%;">
         <col style="width: 7%;">
         <col style="width: 7%;">
+        <col style="width: 7%;">
         <col style="background-color: rgba(80, 15, 213, 0.5); width: 7%;">
     </colgroup>
     <thead>
@@ -90,6 +149,7 @@ We report in the following table our internal pipeline benchmarks.
             <th>Llama-3.2-1B</th>
             <th>Qwen2.5-1.5B</th>
             <th>SmolLM2-1.7B</th>
+            <th>gemma-2-2b</th>
             <th>Falcon3-1B-Base</th>
         </tr>
     </thead>
@@ -98,116 +158,114 @@ We report in the following table our internal pipeline benchmarks.
             <td rowspan="3">General</td>
             <td>MMLU (5-shot)</td>
             <td>31.1</td>
-            <td><b>61.0</b></td>
-            <td>50.1</td>
+            <td>61.0</td>
+            <td>50.2</td>
+            <td>53.1</td>
             <td>42.5</td>
         </tr>
         <tr>
             <td>MMLU-PRO (5-shot)</td>
             <td>11.7</td>
-            <td><b>28.4</b></td>
-            <td>21.3</td>
-            <td>16.1</td>
+            <td>28.5</td>
+            <td>21.4</td>
+            <td>22.1</td>
+            <td>16.2</td>
         </tr>
         <tr>
             <td>IFEval</td>
-            <td>14.8</td>
-            <td><b>26.0</b></td>
+            <td>14.9</td>
+            <td>26.1</td>
             <td>24.2</td>
-            <td>25.2</td>
+            <td>20.4</td>
+            <td>25.3</td>
         </tr>
         <tr>
             <td rowspan="2">Math</td>
             <td>GSM8K (5-shot)</td>
             <td>6.6</td>
-            <td><b>62.2</b></td>
-            <td>31.0</td>
+            <td>62.3</td>
+            <td>31.1</td>
+            <td>25.6</td>
             <td>34.3</td>
         </tr>
         <tr>
-            <td>MATH Lvl-5 (4-shot)</td>
-            <td>0.2</td>
-            <td><b>6.7</b></td>
-            <td>1.4</td>
+            <td>MATH (4-shot)</td>
+            <td>0.3</td>
+            <td>6.8</td>
+            <td>1.5</td>
+            <td>2.6</td>
             <td>2.2</td>
         </tr>
         <tr>
             <td rowspan="4">Reasoning</td>
             <td>Arc Challenge (25-shot)</td>
             <td>40.2</td>
-            <td><b>54.8</b></td>
+            <td>54.8</td>
             <td>54.1</td>
-            <td>48.1</td>
+            <td>53.7</td>
+            <td>48.2</td>
         </tr>
         <tr>
             <td>GPQA (0-shot)</td>
-            <td>24.2</td>
-            <td>28.1</td>
-            <td><b>28.9</b></td>
+            <td>24.3</td>
+            <td>28.2</td>
+            <td>28.9</td>
+            <td>25.5</td>
             <td>28.1</td>
         </tr>
         <tr>
             <td>MUSR (0-shot)</td>
             <td>34.5</td>
             <td>35.5</td>
-            <td>34.7</td>
-            <td><b>41.9</b></td>
+            <td>34.8</td>
+            <td>42.8</td>
+            <td>41.9</td>
         </tr>
         <tr>
             <td>BBH (3-shot)</td>
             <td>31.2</td>
-            <td><b>41.1</b></td>
-            <td>34.2</td>
-            <td>36.0</td>
+            <td>41.1</td>
+            <td>34.3</td>
+            <td>36.8</td>
+            <td>36.1</td>
         </tr>
         <tr>
             <td rowspan="4">CommonSense Understanding</td>
             <td>PIQA (0-shot)</td>
-            <td>74.5</td>
+            <td>74.6</td>
             <td>76.0</td>
-            <td><b>77.5</b></td>
+            <td>77.5</td>
+            <td>79.2</td>
             <td>74.5</td>
         </tr>
         <tr>
             <td>SciQ (0-shot)</td>
             <td>88.5</td>
-            <td><b>93.1</b></td>
+            <td>93.1</td>
             <td>90.8</td>
+            <td>95.7</td>
             <td>91.1</td>
         </tr>
         <tr>
             <td>Winogrande (0-shot)</td>
             <td>60.4</td>
             <td>63.0</td>
-            <td><b>66.1</b></td>
+            <td>66.1</td>
+            <td>68.6</td>
             <td>61.2</td>
         </tr>
         <tr>
             <td>OpenbookQA (0-shot)</td>
             <td>37.4</td>
             <td>40.4</td>
-            <td><b>44.0</b></td>
+            <td>44.0</td>
+            <td>41.8</td>
             <td>41.0</td>
         </tr>
     </tbody>
 </table>
 
-## Useful links
-- View our [release blogpost](https://huggingface.co/blog/falcon3).
-- Feel free to join [our discord server](https://discord.gg/fwXpMyGc) if you have any questions or to interact with our researchers and developers.
 
-## 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}
-}
-```
+
+# Citation