alexmarques commited on
Commit
a46858f
·
verified ·
1 Parent(s): 9069d04

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +596 -0
README.md ADDED
@@ -0,0 +1,596 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ language:
4
+ - en
5
+ - fr
6
+ - it
7
+ - pt
8
+ - hi
9
+ - es
10
+ - th
11
+ - de
12
+ base_model:
13
+ - meta-llama/Llama-3.1-70B-Instruct
14
+ tags:
15
+ - facebook
16
+ - meta
17
+ - pytorch
18
+ - llama
19
+ - llama-3
20
+ - fp8
21
+ - quantized
22
+ license: llama3.3
23
+ ---
24
+
25
+ # Llama-3.3-70B-Instruct-FP8-dynamic
26
+
27
+ ## Model Overview
28
+ - **Model Architecture:** Meta-Llama-3.1
29
+ - **Input:** Text
30
+ - **Output:** Text
31
+ - **Model Optimizations:**
32
+ - **Weight quantization:** FP8
33
+ - **Activation quantization:** FP8
34
+ - **Intended Use Cases:** Intended for commercial and research use in multiple languages. Instruction tuned text only models are intended for assistant-like chat, whereas pretrained models can be adapted for a variety of natural language generation tasks. The Llama 3.3 model also supports the ability to leverage the outputs of its models to improve other models including synthetic data generation and distillation. The Llama 3.3 Community License allows for these use cases.
35
+ - **Out-of-scope:** Use in any manner that violates applicable laws or regulations (including trade compliance laws). Use in any other way that is prohibited by the Acceptable Use Policy and Llama 3.3 Community License. Use in languages beyond English, German, French, Italian, Portuguese, Hindi, Spanish, and Thai.
36
+ - **Release Date:** 12/11/2024
37
+ - **Version:** 1.0
38
+ - **License(s):** llama3.3
39
+ - **Model Developers:** RedHat (Neural Magic)
40
+
41
+ ### Model Optimizations
42
+
43
+ This model was obtained by quantizing activation and weights of [Llama-3.3-70B-Instruct](https://huggingface.co/meta-llama/Llama-3.3-70B-Instruct) to FP8 data type.
44
+ This optimization reduces the number of bits used to represent weights and activations from 16 to 8, reducing GPU memory requirements (by approximately 50%) and increasing matrix-multiply compute throughput (by approximately 2x).
45
+ Weight quantization also reduces disk size requirements by approximately 50%.
46
+
47
+ Only weights and activations of the linear operators within transformers blocks are quantized.
48
+ Weights are quantized with a symmetric static per-channel scheme, whereas activations are quantized with a symmetric dynamic per-token scheme.
49
+ The [llm-compressor](https://github.com/vllm-project/llm-compressor) library is used for quantization.
50
+
51
+ ## Deployment
52
+
53
+ This model can be deployed efficiently using the [vLLM](https://docs.vllm.ai/en/latest/) backend, as shown in the example below.
54
+
55
+ ```python
56
+ from vllm import LLM, SamplingParams
57
+ from transformers import AutoTokenizer
58
+
59
+ model_id = "neuralmagic-ent/phi-4-FP8-dynamic"
60
+ number_gpus = 1
61
+
62
+ sampling_params = SamplingParams(temperature=0.7, top_p=0.8, max_tokens=256)
63
+
64
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
65
+
66
+ prompt = "Give me a short introduction to large language model."
67
+
68
+ llm = LLM(model=model_id, tensor_parallel_size=number_gpus)
69
+
70
+ outputs = llm.generate(prompt, sampling_params)
71
+
72
+ generated_text = outputs[0].outputs[0].text
73
+ print(generated_text)
74
+ ```
75
+
76
+ vLLM aslo supports OpenAI-compatible serving. See the [documentation](https://docs.vllm.ai/en/latest/) for more details.
77
+
78
+ ## Creation
79
+
80
+ <details>
81
+ <summary>Creation details</summary>
82
+ This model was created with [llm-compressor](https://github.com/vllm-project/llm-compressor) by running the code snippet below.
83
+
84
+
85
+ ```python
86
+ from transformers import AutoModelForCausalLM, AutoTokenizer
87
+ from llmcompressor.modifiers.quantization import QuantizationModifier
88
+ from llmcompressor.transformers import oneshot
89
+
90
+ # Load model
91
+ model_stub = "meta-llama/Llama-3.3-70B-Instruct"
92
+ model_name = model_stub.split("/")[-1]
93
+
94
+ tokenizer = AutoTokenizer.from_pretrained(model_stub)
95
+
96
+ model = AutoModelForCausalLM.from_pretrained(
97
+ model_stub,
98
+ device_map="auto",
99
+ torch_dtype="auto",
100
+ )
101
+
102
+ # Configure the quantization algorithm and scheme
103
+ recipe = QuantizationModifier(
104
+ targets="Linear",
105
+ scheme="FP8_dynamic",
106
+ ignore=["lm_head"],
107
+ )
108
+
109
+ # Apply quantization
110
+ oneshot(
111
+ model=model,
112
+ recipe=recipe,
113
+ )
114
+
115
+ # Save to disk in compressed-tensors format
116
+ save_path = model_name + "-FP8-dynamic"
117
+ model.save_pretrained(save_path)
118
+ tokenizer.save_pretrained(save_path)
119
+ print(f"Model and tokenizer saved to: {save_path}")
120
+ ```
121
+ </details>
122
+
123
+ ## Evaluation
124
+
125
+ This model was evaluated on the well-known OpenLLM v1, OpenLLM v2, HumanEval, and HumanEval+ benchmarks.
126
+ In all cases, model outputs were generated with the [vLLM](https://docs.vllm.ai/en/stable/) engine.
127
+
128
+ OpenLLM v1 and v2 evaluations were conducted using [lm-evaluation-harness](https://github.com/EleutherAI/lm-evaluation-harness) and the prompting style of [Meta-Llama-3.1-Instruct-evals](https://huggingface.co/datasets/meta-llama/Meta-Llama-3.1-8B-Instruct-evals) when available.
129
+
130
+ HumanEval and HumanEval+ evaluations were conducted using Neural Magic's fork of the [EvalPlus](https://github.com/neuralmagic/evalplus) repository.
131
+
132
+ <details>
133
+ <summary>Evaluation details</summary>
134
+
135
+ **MMLU**
136
+ ```
137
+ lm_eval \
138
+ --model vllm \
139
+ --model_args pretrained="neuralmagic-ent/Llama-3.3-70B-Instruct-FP8-dynamic",dtype=auto,max_model_len=3850,max_gen_toks=10,tensor_parallel_size=1 \
140
+ --tasks mmlu_llama \
141
+ --fewshot_as_multiturn \
142
+ --apply_chat_template \
143
+ --num_fewshot 5 \
144
+ --batch_size auto
145
+ ```
146
+
147
+ **MMLU-CoT**
148
+ ```
149
+ lm_eval \
150
+ --model vllm \
151
+ --model_args pretrained="neuralmagic-ent/Llama-3.3-70B-Instruct-FP8-dynamic",dtype=auto,max_model_len=4064,max_gen_toks=1024,tensor_parallel_size=1 \
152
+ --tasks mmlu_cot_llama \
153
+ --apply_chat_template \
154
+ --num_fewshot 0 \
155
+ --batch_size auto
156
+ ```
157
+
158
+ **ARC-Challenge**
159
+ ```
160
+ lm_eval \
161
+ --model vllm \
162
+ --model_args pretrained="neuralmagic-ent/Llama-3.3-70B-Instruct-FP8-dynamic",dtype=auto,max_model_len=3940,max_gen_toks=100,tensor_parallel_size=1 \
163
+ --tasks arc_challenge_llama \
164
+ --apply_chat_template \
165
+ --num_fewshot 0 \
166
+ --batch_size auto
167
+ ```
168
+
169
+ **GSM-8K**
170
+ ```
171
+ lm_eval \
172
+ --model vllm \
173
+ --model_args pretrained="neuralmagic-ent/Llama-3.3-70B-Instruct-FP8-dynamic",dtype=auto,max_model_len=4096,max_gen_toks=1024,tensor_parallel_size=1 \
174
+ --tasks gsm8k_llama \
175
+ --fewshot_as_multiturn \
176
+ --apply_chat_template \
177
+ --num_fewshot 8 \
178
+ --batch_size auto
179
+ ```
180
+
181
+ **Hellaswag**
182
+ ```
183
+ lm_eval \
184
+ --model vllm \
185
+ --model_args pretrained="neuralmagic-ent/Llama-3.3-70B-Instruct-FP8-dynamic",dtype=auto,add_bos_token=True,max_model_len=4096,tensor_parallel_size=1 \
186
+ --tasks hellaswag \
187
+ --num_fewshot 10 \
188
+ --batch_size auto
189
+ ```
190
+
191
+ **Winogrande**
192
+ ```
193
+ lm_eval \
194
+ --model vllm \
195
+ --model_args pretrained="neuralmagic-ent/Llama-3.3-70B-Instruct-FP8-dynamic",dtype=auto,add_bos_token=True,max_model_len=4096,tensor_parallel_size=1 \
196
+ --tasks winogrande \
197
+ --num_fewshot 5 \
198
+ --batch_size auto
199
+ ```
200
+
201
+ **TruthfulQA**
202
+ ```
203
+ lm_eval \
204
+ --model vllm \
205
+ --model_args pretrained="neuralmagic-ent/Llama-3.3-70B-Instruct-FP8-dynamic",dtype=auto,add_bos_token=True,max_model_len=4096,tensor_parallel_size=1 \
206
+ --tasks truthfulqa \
207
+ --num_fewshot 0 \
208
+ --batch_size auto
209
+ ```
210
+
211
+ **OpenLLM v2**
212
+ ```
213
+ lm_eval \
214
+ --model vllm \
215
+ --model_args pretrained="neuralmagic-ent/Llama-3.3-70B-Instruct-FP8-dynamic",dtype=auto,max_model_len=4096,tensor_parallel_size=1,enable_chunked_prefill=True \
216
+ --apply_chat_template \
217
+ --fewshot_as_multiturn \
218
+ --tasks leaderboard \
219
+ --batch_size auto
220
+ ```
221
+
222
+ **MMLU Portuguese**
223
+ ```
224
+ lm_eval \
225
+ --model vllm \
226
+ --model_args pretrained="neuralmagic-ent/Llama-3.3-70B-Instruct-FP8-dynamic",dtype=auto,max_model_len=3850,max_gen_toks=10,tensor_parallel_size=1 \
227
+ --tasks mmlu_pt_llama \
228
+ --fewshot_as_multiturn \
229
+ --apply_chat_template \
230
+ --num_fewshot 5 \
231
+ --batch_size auto
232
+ ```
233
+
234
+ **MMLU Spanish**
235
+ ```
236
+ lm_eval \
237
+ --model vllm \
238
+ --model_args pretrained="neuralmagic-ent/Llama-3.3-70B-Instruct-FP8-dynamic",dtype=auto,max_model_len=3850,max_gen_toks=10,tensor_parallel_size=1 \
239
+ --tasks mmlu_es_llama \
240
+ --fewshot_as_multiturn \
241
+ --apply_chat_template \
242
+ --num_fewshot 5 \
243
+ --batch_size auto
244
+ ```
245
+
246
+ **MMLU Italian**
247
+ ```
248
+ lm_eval \
249
+ --model vllm \
250
+ --model_args pretrained="neuralmagic-ent/Llama-3.3-70B-Instruct-FP8-dynamic",dtype=auto,max_model_len=3850,max_gen_toks=10,tensor_parallel_size=1 \
251
+ --tasks mmlu_it_llama \
252
+ --fewshot_as_multiturn \
253
+ --apply_chat_template \
254
+ --num_fewshot 5 \
255
+ --batch_size auto
256
+ ```
257
+
258
+ **MMLU German**
259
+ ```
260
+ lm_eval \
261
+ --model vllm \
262
+ --model_args pretrained="neuralmagic-ent/Llama-3.3-70B-Instruct-FP8-dynamic",dtype=auto,max_model_len=3850,max_gen_toks=10,tensor_parallel_size=1 \
263
+ --tasks mmlu_de_llama \
264
+ --fewshot_as_multiturn \
265
+ --apply_chat_template \
266
+ --num_fewshot 5 \
267
+ --batch_size auto
268
+ ```
269
+
270
+ **MMLU French**
271
+ ```
272
+ lm_eval \
273
+ --model vllm \
274
+ --model_args pretrained="neuralmagic-ent/Llama-3.3-70B-Instruct-FP8-dynamic",dtype=auto,max_model_len=3850,max_gen_toks=10,tensor_parallel_size=1 \
275
+ --tasks mmlu_fr_llama \
276
+ --fewshot_as_multiturn \
277
+ --apply_chat_template \
278
+ --num_fewshot 5 \
279
+ --batch_size auto
280
+ ```
281
+
282
+ **MMLU Hindi**
283
+ ```
284
+ lm_eval \
285
+ --model vllm \
286
+ --model_args pretrained="neuralmagic-ent/Llama-3.3-70B-Instruct-FP8-dynamic",dtype=auto,max_model_len=3850,max_gen_toks=10,tensor_parallel_size=1 \
287
+ --tasks mmlu_hi_llama \
288
+ --fewshot_as_multiturn \
289
+ --apply_chat_template \
290
+ --num_fewshot 5 \
291
+ --batch_size auto
292
+ ```
293
+
294
+ **MMLU Thai**
295
+ ```
296
+ lm_eval \
297
+ --model vllm \
298
+ --model_args pretrained="neuralmagic-ent/Llama-3.3-70B-Instruct-FP8-dynamic",dtype=auto,max_model_len=3850,max_gen_toks=10,tensor_parallel_size=1 \
299
+ --tasks mmlu_th_llama \
300
+ --fewshot_as_multiturn \
301
+ --apply_chat_template \
302
+ --num_fewshot 5 \
303
+ --batch_size auto
304
+ ```
305
+
306
+ **HumanEval and HumanEval+**
307
+ *Generation*
308
+ ```
309
+ python3 codegen/generate.py \
310
+ --model neuralmagic-ent/Llama-3.3-70B-Instruct-FP8-dynamic \
311
+ --bs 16 \
312
+ --temperature 0.2 \
313
+ --n_samples 50 \
314
+ --root "." \
315
+ --dataset humaneval
316
+ ```
317
+
318
+ *Sanitization*
319
+ ```
320
+ python3 evalplus/sanitize.py \
321
+ humaneval/neuralmagic-ent--Llama-3.3-70B-Instruct-FP8-dynamic_vllm_temp_0.2
322
+ ```
323
+
324
+ *Evaluation*
325
+ ```
326
+ evalplus.evaluate \
327
+ --dataset humaneval \
328
+ --samples humaneval/neuralmagic-ent--Llama-3.3-70B-Instruct-FP8-dynamic_vllm_temp_0.2-sanitized
329
+ ```
330
+
331
+ ### Accuracy
332
+
333
+ <table>
334
+ <tr>
335
+ <th>Category
336
+ </th>
337
+ <th>Benchmark
338
+ </th>
339
+ <th>Llama-3.3-70B-Instruct
340
+ </th>
341
+ <th>Llama-3.3-70B-Instruct-FP8-dynamic<br>(this model)
342
+ </th>
343
+ <th>Recovery
344
+ </th>
345
+ </tr>
346
+ <tr>
347
+ <td rowspan="8" ><strong>OpenLLM v1</strong>
348
+ </td>
349
+ <td>MMLU (5-shot)
350
+ </td>
351
+ <td>81.60
352
+ </td>
353
+ <td>81.31
354
+ </td>
355
+ <td>99.6%
356
+ </td>
357
+ </tr>
358
+ <tr>
359
+ <td>MMLU (CoT, 0-shot)
360
+ </td>
361
+ <td>86.58
362
+ </td>
363
+ <td>86.34
364
+ </td>
365
+ <td>99.7%
366
+ </td>
367
+ </tr>
368
+ <tr>
369
+ <td>ARC Challenge (0-shot)
370
+ </td>
371
+ <td>49.23
372
+ </td>
373
+ <td>51.96
374
+ </td>
375
+ <td>105.6%
376
+ </td>
377
+ </tr>
378
+ <tr>
379
+ <td>GSM-8K (CoT, 8-shot, strict-match)
380
+ </td>
381
+ <td>94.16
382
+ </td>
383
+ <td>94.92
384
+ </td>
385
+ <td>100.8%
386
+ </td>
387
+ </tr>
388
+ <tr>
389
+ <td>Hellaswag (10-shot)
390
+ </td>
391
+ <td>86.49
392
+ </td>
393
+ <td>86.43
394
+ </td>
395
+ <td>99.9%
396
+ </td>
397
+ </tr>
398
+ <tr>
399
+ <td>Winogrande (5-shot)
400
+ </td>
401
+ <td>84.77
402
+ </td>
403
+ <td>84.53
404
+ </td>
405
+ <td>99.7%
406
+ </td>
407
+ </tr>
408
+ <tr>
409
+ <td>TruthfulQA (0-shot, mc2)
410
+ </td>
411
+ <td>62.75
412
+ </td>
413
+ <td>63.21
414
+ </td>
415
+ <td>100.7%
416
+ </td>
417
+ </tr>
418
+ <tr>
419
+ <td><strong>Average</strong>
420
+ </td>
421
+ <td><strong>77.94</strong>
422
+ </td>
423
+ <td><strong>78.39</strong>
424
+ </td>
425
+ <td><strong>100.6%</strong>
426
+ </td>
427
+ </tr>
428
+ <tr>
429
+ <td rowspan="7" ><strong>OpenLLM v2</strong>
430
+ </td>
431
+ <td>MMLU-Pro (5-shot)
432
+ </td>
433
+ <td>51.89
434
+ </td>
435
+ <td>51.50
436
+ </td>
437
+ <td>99.3%
438
+ </td>
439
+ </tr>
440
+ <tr>
441
+ <td>IFEval (0-shot)
442
+ </td>
443
+ <td>90.89
444
+ </td>
445
+ <td>90.92
446
+ </td>
447
+ <td>100.0%
448
+ </td>
449
+ </tr>
450
+ <tr>
451
+ <td>BBH (3-shot)
452
+ </td>
453
+ <td>63.15
454
+ </td>
455
+ <td>62.84
456
+ </td>
457
+ <td>99.5%
458
+ </td>
459
+ </tr>
460
+ <tr>
461
+ <td>Math-lvl-5 (4-shot)
462
+ </td>
463
+ <td>0.17
464
+ </td>
465
+ <td>0.33
466
+ </td>
467
+ <td>N/A
468
+ </td>
469
+ </tr>
470
+ <tr>
471
+ <td>GPQA (0-shot)
472
+ </td>
473
+ <td>46.10
474
+ </td>
475
+ <td>46.30
476
+ </td>
477
+ <td>100.4%
478
+ </td>
479
+ </tr>
480
+ <tr>
481
+ <td>MuSR (0-shot)
482
+ </td>
483
+ <td>44.35
484
+ </td>
485
+ <td>43.96
486
+ </td>
487
+ <td>99.1%
488
+ </td>
489
+ </tr>
490
+ <tr>
491
+ <td><strong>Average</strong>
492
+ </td>
493
+ <td><strong>49.42</strong>
494
+ </td>
495
+ <td><strong>49.31</strong>
496
+ </td>
497
+ <td><strong>99.8%</strong>
498
+ </td>
499
+ </tr>
500
+ <tr>
501
+ <td rowspan="2" ><strong>Coding</strong>
502
+ </td>
503
+ <td>HumanEval pass@1
504
+ </td>
505
+ <td>83.20
506
+ </td>
507
+ <td>83.70
508
+ </td>
509
+ <td>100.6%
510
+ </td>
511
+ </tr>
512
+ <tr>
513
+ <td>HumanEval+ pass@1
514
+ </td>
515
+ <td>78.40
516
+ </td>
517
+ <td>78.70
518
+ </td>
519
+ <td>100.4%
520
+ </td>
521
+ </tr>
522
+ <tr>
523
+ <td rowspan="9" ><strong>Multilingual</strong>
524
+ </td>
525
+ <td>Portuguese MMLU (5-shot)
526
+ </td>
527
+ <td>79.76
528
+ </td>
529
+ <td>79.75
530
+ </td>
531
+ <td>100.0%
532
+ </td>
533
+ </tr>
534
+ <tr>
535
+ <td>Spanish MMLU (5-shot)
536
+ </td>
537
+ <td>79.33
538
+ </td>
539
+ <td>79.17
540
+ </td>
541
+ <td>99.8%
542
+ </td>
543
+ </tr>
544
+ <tr>
545
+ <td>Italian MMLU (5-shot)
546
+ </td>
547
+ <td>79.15
548
+ </td>
549
+ <td>78.84
550
+ </td>
551
+ <td>99.6%
552
+ </td>
553
+ </tr>
554
+ <tr>
555
+ <td>German MMLU (5-shot)
556
+ </td>
557
+ <td>77.94
558
+ </td>
559
+ <td>77.95
560
+ </td>
561
+ <td>100.0%
562
+ </td>
563
+ </tr>
564
+ <tr>
565
+ <td>French MMLU (5-shot)
566
+ </td>
567
+ <td>75.69
568
+ </td>
569
+ <td>75.45
570
+ </td>
571
+ <td>99.7%
572
+ </td>
573
+ </tr>
574
+ <tr>
575
+ <td>Hindi MMLU (5-shot)
576
+ </td>
577
+ <td>73.81
578
+ </td>
579
+ <td>73.71
580
+ </td>
581
+ <td>99.9%
582
+ </td>
583
+ </tr>
584
+ <tr>
585
+ <td>Thai MMLU (5-shot)
586
+ </td>
587
+ <td>71.98
588
+ </td>
589
+ <td>71.77
590
+ </td>
591
+ <td>99.7%
592
+ </td>
593
+ </tr>
594
+ </table>
595
+
596
+