lucyknada commited on
Commit
a13beb7
·
verified ·
1 Parent(s): 06bf137

Upload ./README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +378 -0
README.md ADDED
@@ -0,0 +1,378 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ pipeline_tag: text-generation
3
+ inference: false
4
+ license: apache-2.0
5
+ library_name: transformers
6
+ tags:
7
+ - language
8
+ - granite-3.2
9
+ base_model:
10
+ - ibm-granite/granite-3.1-8b-instruct
11
+ ---
12
+ ### exl2 quant (measurement.json in main branch)
13
+ ---
14
+ ### check revisions for quants
15
+ ---
16
+
17
+
18
+ # Granite-3.2-8B-Instruct
19
+
20
+ **Model Summary:**
21
+ Granite-3.2-8B-Instruct is an 8-billion-parameter, long-context AI model fine-tuned for thinking capabilities. Built on top of [Granite-3.1-8B-Instruct](https://huggingface.co/ibm-granite/granite-3.1-8b-instruct), it has been trained using a mix of permissively licensed open-source datasets and internally generated synthetic data designed for reasoning tasks. The model allows controllability of its thinking capability, ensuring it is applied only when required.
22
+
23
+
24
+ - **Developers:** Granite Team, IBM
25
+ - **Website**: [Granite Docs](https://www.ibm.com/granite/docs/)
26
+ - **Release Date**: February 26th, 2025
27
+ - **License:** [Apache 2.0](https://www.apache.org/licenses/LICENSE-2.0)
28
+
29
+ **Supported Languages:**
30
+ English, German, Spanish, French, Japanese, Portuguese, Arabic, Czech, Italian, Korean, Dutch, and Chinese. However, users may finetune this Granite model for languages beyond these 12 languages.
31
+
32
+ **Intended Use:**
33
+ This model is designed to handle general instruction-following tasks and can be integrated into AI assistants across various domains, including business applications.
34
+
35
+ **Capabilities**
36
+ * **Thinking**
37
+ * Summarization
38
+ * Text classification
39
+ * Text extraction
40
+ * Question-answering
41
+ * Retrieval Augmented Generation (RAG)
42
+ * Code related tasks
43
+ * Function-calling tasks
44
+ * Multilingual dialog use cases
45
+ * Long-context tasks including long document/meeting summarization, long document QA, etc.
46
+
47
+
48
+
49
+ **Generation:**
50
+ This is a simple example of how to use Granite-3.2-8B-Instruct model.
51
+
52
+ Install the following libraries:
53
+
54
+ ```shell
55
+ pip install torch torchvision torchaudio
56
+ pip install accelerate
57
+ pip install transformers
58
+ ```
59
+ Then, copy the snippet from the section that is relevant for your use case.
60
+
61
+ ```python
62
+ from transformers import AutoModelForCausalLM, AutoTokenizer, set_seed
63
+ import torch
64
+
65
+ model_path="ibm-granite/granite-3.2-8b-instruct"
66
+ device="cuda"
67
+ model = AutoModelForCausalLM.from_pretrained(
68
+ model_path,
69
+ device_map=device,
70
+ torch_dtype=torch.bfloat16,
71
+ )
72
+ tokenizer = AutoTokenizer.from_pretrained(
73
+ model_path
74
+ )
75
+
76
+ conv = [{"role": "user", "content":"You have 10 liters of a 30% acid solution. How many liters of a 70% acid solution must be added to achieve a 50% acid mixture?"}]
77
+
78
+ input_ids = tokenizer.apply_chat_template(conv, return_tensors="pt", thinking=True, return_dict=True, add_generation_prompt=True).to(device)
79
+
80
+ set_seed(42)
81
+ output = model.generate(
82
+ **input_ids,
83
+ max_new_tokens=8192,
84
+ )
85
+
86
+ prediction = tokenizer.decode(output[0, input_ids["input_ids"].shape[1]:], skip_special_tokens=True)
87
+ print(prediction)
88
+ ```
89
+
90
+ **Example Outputs**
91
+ - thinking=True
92
+ ```md
93
+ Here is my thought process:
94
+
95
+ 1. We know we start with 10 liters of a 30% acid solution, so there's 3 liters of pure acid in it (10 liters * 0.3).
96
+ 2. We need to determine how much of the 70% acid solution to add such that the final mixture is 50% acid.
97
+ 3. Let's say x liters of the 70% acid solution are added. This will contribute 0.7x liters of pure acid.
98
+ 4. The total volume of the new mixture will be 10 + x liters, and the total amount of pure acid will be 3 + 0.7x liters.
99
+ 5. We need to set up an equation where (3 + 0.7x) / (10 + x) = 0.5 (since we want a 50% acid mixture).
100
+ 6. Solving this equation for x will give us the volume of 70% solution needed.
101
+
102
+ Here is my response:
103
+
104
+ To solve this, we need to set up an equation based on the concentration of acid before and after mixing.
105
+
106
+ 1. **Initial Acid Content**:
107
+ - We start with 10 liters of a 30% acid solution.
108
+ - Pure acid in this = 10 * 0.3 = 3 liters.
109
+
110
+ 2. **Variable for Addition**:
111
+ - Let x be the liters of 70% acid solution to be added.
112
+ - Pure acid in x liters of 70% solution = 0.7x liters.
113
+
114
+ 3. **Final Mixture Concentration**:
115
+ - Total volume of mixture = 10 + x liters.
116
+ - Total pure acid in the mixture = 3 (from initial solution) + 0.7x (from added solution).
117
+
118
+ 4. **Setting up the equation for 50% acid concentration**:
119
+ - We require that the ratio of pure acid to total volume is 0.5 (50%).
120
+ - Therefore, the equation is:
121
+
122
+ \[
123
+ \frac{3 + 0.7x}{10 + x} = 0.5
124
+ \]
125
+
126
+ 5. **Solving the equation**:
127
+
128
+ Multiply both sides by (10 + x) to clear the fraction:
129
+
130
+ \[
131
+ 3 + 0.7x = 0.5(10 + x)
132
+ \]
133
+
134
+ Expanding the right side:
135
+
136
+ \[
137
+ 3 + 0.7x = 5 + 0.5x
138
+ \]
139
+
140
+ Subtract 0.5x from both sides:
141
+
142
+ \[
143
+ 3 + 0.2x = 5
144
+ \]
145
+
146
+ Subtract 3 from both sides:
147
+
148
+ \[
149
+ 0.2x = 2
150
+ \]
151
+
152
+ Divide by 0.2:
153
+
154
+ \[
155
+ x = 10
156
+ \]
157
+
158
+ So, 10 liters of the 70% acid solution must be added to the initial 10 liters of 30% solution to get a 50% acid mixture.
159
+
160
+ Let's verify:
161
+
162
+ - Initial acid: 3 liters
163
+ - Acid from added solution: 0.7 * 10 = 7 liters
164
+ - Total acid: 3 + 7 = 10 liters
165
+ - Total volume: 10 + 10 = 20 liters
166
+ - Acid percentage: \(\frac{10}{20} = 0.5 = 50\%\)
167
+
168
+ This confirms our solution.
169
+
170
+ Final answer: 10 liters of 70% acid solution should be added.
171
+ ```
172
+ - thinking=False
173
+ ```md
174
+ Let's denote the amount of 70% acid solution to be added as x liters.
175
+
176
+ We start with 10 liters of a 30% acid solution, which means there are 0.3 * 10 = 3 liters of pure acid in it.
177
+
178
+ We want to end up with a 50% acid solution after adding x liters of 70% acid solution. The total volume of the mixture will be 10 + x liters, and we want half of this volume to be pure acid.
179
+
180
+ So, the amount of pure acid in the final mixture should be 0.5 * (10 + x) liters.
181
+
182
+ The amount of pure acid in the final mixture comes from two sources: the original 30% solution and the added 70% solution. So, we can set up the following equation:
183
+
184
+ 3 (from the original solution) + 0.7x (from the added solution) = 0.5 * (10 + x)
185
+
186
+ Now, let's solve for x:
187
+
188
+ 3 + 0.7x = 5 + 0.5x
189
+ 0.7x - 0.5x = 5 - 3
190
+ 0.2x = 2
191
+ x = 2 / 0.2
192
+ x = 10
193
+
194
+ So, you need to add 10 liters of a 70% acid solution to the 10 liters of a 30% acid solution to get a 50% acid mixture.
195
+ ```
196
+
197
+ **Evaluation Results:**
198
+ <table>
199
+
200
+ <thead>
201
+ <tr>
202
+ <th style="text-align:left; background-color: #001d6c; color: white;">Models</th>
203
+ <th style="text-align:center; background-color: #001d6c; color: white;">ArenaHard</th>
204
+ <th style="text-align:center; background-color: #001d6c; color: white;">Alpaca-Eval-2</th>
205
+ <th style="text-align:center; background-color: #001d6c; color: white;">MMLU</th>
206
+ <th style="text-align:center; background-color: #001d6c; color: white;">PopQA</th>
207
+ <th style="text-align:center; background-color: #001d6c; color: white;">TruthfulQA</th>
208
+ <th style="text-align:center; background-color: #001d6c; color: white;">BigBenchHard</th>
209
+ <th style="text-align:center; background-color: #001d6c; color: white;">DROP</th>
210
+ <th style="text-align:center; background-color: #001d6c; color: white;">GSM8K</th>
211
+ <th style="text-align:center; background-color: #001d6c; color: white;">HumanEval</th>
212
+ <th style="text-align:center; background-color: #001d6c; color: white;">HumanEval+</th>
213
+ <th style="text-align:center; background-color: #001d6c; color: white;">IFEval</th>
214
+ <th style="text-align:center; background-color: #001d6c; color: white;">AttaQ</th>
215
+ </tr></thead>
216
+ <tbody>
217
+ <tr>
218
+ <td style="text-align:left; background-color: #DAE8FF; color: black;">Llama-3.1-8B-Instruct</td>
219
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">36.43</td>
220
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">27.22</td>
221
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">69.15</td>
222
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">28.79</td>
223
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">52.79</td>
224
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">72.66</td>
225
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">61.48</td>
226
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">83.24</td>
227
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">85.32</td>
228
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">80.15</td>
229
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">79.10</td>
230
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">83.43</td>
231
+ </tr>
232
+
233
+ <tr>
234
+ <td style="text-align:left; background-color: #DAE8FF; color: black;">DeepSeek-R1-Distill-Llama-8B</td>
235
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">17.17</td>
236
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">21.85</td>
237
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">45.80</td>
238
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">13.25</td>
239
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">47.43</td>
240
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">65.71</td>
241
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">44.46</td>
242
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">72.18</td>
243
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">67.54</td>
244
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">62.91</td>
245
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">66.50</td>
246
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">42.87</td>
247
+ </tr>
248
+
249
+ <tr>
250
+ <td style="text-align:left; background-color: #DAE8FF; color: black;">Qwen-2.5-7B-Instruct</td>
251
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">25.44</td>
252
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">30.34</td>
253
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">74.30</td>
254
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">18.12</td>
255
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">63.06</td>
256
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">70.40</td>
257
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">54.71</td>
258
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">84.46</td>
259
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">93.35</td>
260
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">89.91</td>
261
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">74.90</td>
262
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">81.90</td>
263
+ </tr>
264
+
265
+ <tr>
266
+ <td style="text-align:left; background-color: #DAE8FF; color: black;">DeepSeek-R1-Distill-Qwen-7B</td>
267
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">10.36</td>
268
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">15.35</td>
269
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">50.72</td>
270
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">9.94</td>
271
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">47.14</td>
272
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">65.04</td>
273
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">42.76</td>
274
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">78.47</td>
275
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">79.89</td>
276
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">78.43</td>
277
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">59.10</td>
278
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">42.45</td>
279
+ </tr>
280
+
281
+ <tr>
282
+ <td style="text-align:left; background-color: #DAE8FF; color: black;">Granite-3.1-8B-Instruct</td>
283
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">37.58</td>
284
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">30.34</td>
285
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">66.77</td>
286
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">28.7</td>
287
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">65.84</td>
288
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">68.55</td>
289
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">50.78</td>
290
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">79.15</td>
291
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">89.63</td>
292
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">85.79</td>
293
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">73.20</td>
294
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">85.73</td>
295
+ </tr>
296
+
297
+
298
+ <tr>
299
+ <td style="text-align:left; background-color: #DAE8FF; color: black;">Granite-3.1-2B-Instruct</td>
300
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">23.3</td>
301
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">27.17</td>
302
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">57.11</td>
303
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">20.55</td>
304
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">59.79</td>
305
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">54.46</td>
306
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">18.68</td>
307
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">67.55</td>
308
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">79.45</td>
309
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">75.26</td>
310
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">63.59</td>
311
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">84.7</td>
312
+ </tr>
313
+
314
+
315
+ <tr>
316
+ <td style="text-align:left; background-color: #DAE8FF; color: black;">Granite-3.2-2B-Instruct</td>
317
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">24.86</td>
318
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">34.51</td>
319
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">57.18</td>
320
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">20.56</td>
321
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">59.8</td>
322
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">52.27</td>
323
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">21.12</td>
324
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">67.02</td>
325
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">80.13</td>
326
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">73.39</td>
327
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">61.55</td>
328
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">83.23</td>
329
+ </tr>
330
+
331
+ <tr>
332
+ <td style="text-align:left; background-color: #DAE8FF; color: black;"><b>Granite-3.2-8B-Instruct</b></td>
333
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">55.25</td>
334
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">61.19</td>
335
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">66.79</td>
336
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">28.04</td>
337
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">66.92</td>
338
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">64.77</td>
339
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">50.95</td>
340
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">81.65</td>
341
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">89.35</td>
342
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">85.72</td>
343
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">74.31</td>
344
+ <td style="text-align:center; background-color: #DAE8FF; color: black;">85.42</td>
345
+
346
+ </tr>
347
+
348
+
349
+
350
+ </tbody></table>
351
+
352
+ **Training Data:**
353
+ Overall, our training data is largely comprised of two key sources: (1) publicly available datasets with permissive license, (2) internal synthetically generated data targeted to enhance reasoning capabilites.
354
+ <!-- A detailed attribution of datasets can be found in [Granite 3.2 Technical Report (coming soon)](#), and [Accompanying Author List](https://github.com/ibm-granite/granite-3.0-language-models/blob/main/author-ack.pdf). -->
355
+
356
+ **Infrastructure:**
357
+ We train Granite-3.2-8B-Instruct using IBM's super computing cluster, Blue Vela, which is outfitted with NVIDIA H100 GPUs. This cluster provides a scalable and efficient infrastructure for training our models over thousands of GPUs.
358
+
359
+ **Ethical Considerations and Limitations:**
360
+ Granite-3.2-8B-Instruct builds upon Granite-3.1-8B-Instruct, leveraging both permissively licensed open-source and select proprietary data for enhanced performance. Since it inherits its foundation from the previous model, all ethical considerations and limitations applicable to [Granite-3.1-8B-Instruct](https://huggingface.co/ibm-granite/granite-3.1-8b-instruct) remain relevant.
361
+
362
+
363
+ **Resources**
364
+ - ⭐️ Learn about the latest updates with Granite: https://www.ibm.com/granite
365
+ - 📄 Get started with tutorials, best practices, and prompt engineering advice: https://www.ibm.com/granite/docs/
366
+ - 💡 Learn about the latest Granite learning resources: https://ibm.biz/granite-learning-resources
367
+
368
+ <!-- ## Citation
369
+ ```
370
+ @misc{granite-models,
371
+ author = {author 1, author2, ...},
372
+ title = {},
373
+ journal = {},
374
+ volume = {},
375
+ year = {2024},
376
+ url = {https://arxiv.org/abs/0000.00000},
377
+ }
378
+ ``` -->