nm-research commited on
Commit
c2f5b24
·
verified ·
1 Parent(s): 51f8c7b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +243 -0
README.md CHANGED
@@ -261,3 +261,246 @@ lm_eval \
261
  </tr>
262
  </tbody>
263
  </table>
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
261
  </tr>
262
  </tbody>
263
  </table>
264
+
265
+ ## Inference Performance
266
+
267
+
268
+ This model achieves up to 1.1x speedup in single-stream deployment, depending on hardware and use-case scenario.
269
+ The following performance benchmarks were conducted with [vLLM](https://docs.vllm.ai/en/latest/) version 0.6.7.2, and [GuideLLM](https://github.com/neuralmagic/guidellm).
270
+
271
+ <details>
272
+ <summary>Benchmarking Command</summary>
273
+
274
+ ```
275
+ guidellm --model neuralmagic/DeepSeek-R1-Distill-Qwen-1.5B-quantized.w4a16 --target "http://localhost:8000/v1" --data-type emulated --data "prompt_tokens=<prompt_tokens>,generated_tokens=<generated_tokens>" --max seconds 360 --backend aiohttp_server
276
+ ```
277
+ </details>
278
+
279
+ ### Single-stream performance (measured with vLLM version 0.7.2)
280
+ <table>
281
+ <thead>
282
+ <tr>
283
+ <th></th>
284
+ <th></th>
285
+ <th></th>
286
+ <th style="text-align: center;" colspan="2" >Instruction Following<br>256 / 128</th>
287
+ <th style="text-align: center;" colspan="2" >Multi-turn Chat<br>512 / 256</th>
288
+ <th style="text-align: center;" colspan="2" >Docstring Generation<br>768 / 128</th>
289
+ <th style="text-align: center;" colspan="2" >RAG<br>1024 / 128</th>
290
+ <th style="text-align: center;" colspan="2" >Code Completion<br>256 / 1024</th>
291
+ <th style="text-align: center;" colspan="2" >Code Fixing<br>1024 / 1024</th>
292
+ <th style="text-align: center;" colspan="2" >Large Summarization<br>4096 / 512</th>
293
+ <th style="text-align: center;" colspan="2" >Large RAG<br>10240 / 1536</th>
294
+ </tr>
295
+ <tr>
296
+ <th>Hardware</th>
297
+ <th>Model</th>
298
+ <th>Average cost reduction</th>
299
+ <th>Latency (s)</th>
300
+ <th>QPD</th>
301
+ <th>Latency (s)</th>
302
+ <th>QPD</th>
303
+ <th>Latency (s)</th>
304
+ <th>QPD</th>
305
+ <th>Latency (s)</th>
306
+ <th>QPD</th>
307
+ <th>Latency (s)</th>
308
+ <th>QPD</th>
309
+ <th>Latency (s)</th>
310
+ <th>QPD</th>
311
+ <th>Latency (s)</th>
312
+ <th>QPD</th>
313
+ <th>Latency (s)</th>
314
+ <th>QPD</th>
315
+ </tr>
316
+ </thead>
317
+ <tbody style="text-align: center" >
318
+ <tr>
319
+ <th rowspan="3" valign="top">A6000x1</th>
320
+ <th>deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B</th>
321
+ <td>---</td>
322
+ <td>0.8</td>
323
+ <td>5667</td>
324
+ <td>1.6</td>
325
+ <td>2776</td>
326
+ <td>0.8</td>
327
+ <td>5515</td>
328
+ <td>0.8</td>
329
+ <td>5466</td>
330
+ <td>6.4</td>
331
+ <td>705</td>
332
+ <td>6.5</td>
333
+ <td>697</td>
334
+ <td>3.5</td>
335
+ <td>1295</td>
336
+ <td>18.3</td>
337
+ <td>246</td>
338
+ </tr>
339
+ <tr>
340
+ <th>neuralmagic/DeepSeek-R1-Distill-Qwen-1.5B-quantized.w8a8</th>
341
+ <td>1.14</td>
342
+ <td>0.7</td>
343
+ <td>6635</td>
344
+ <td>1.3</td>
345
+ <td>3340</td>
346
+ <td>0.7</td>
347
+ <td>6396</td>
348
+ <td>0.7</td>
349
+ <td>6343</td>
350
+ <td>5.3</td>
351
+ <td>845</td>
352
+ <td>5.4</td>
353
+ <td>832</td>
354
+ <td>2.9</td>
355
+ <td>1547</td>
356
+ <td>21.3</td>
357
+ <td>211</td>
358
+ </tr>
359
+ <tr>
360
+ <th>neuralmagic/DeepSeek-R1-Distill-Qwen-1.5B-quantized.w4a16</th>
361
+ <td>1.38</td>
362
+ <td>0.5</td>
363
+ <td>8293</td>
364
+ <td>1.1</td>
365
+ <td>4184</td>
366
+ <td>0.6</td>
367
+ <td>7976</td>
368
+ <td>0.6</td>
369
+ <td>7504</td>
370
+ <td>4.3</td>
371
+ <td>1051</td>
372
+ <td>4.4</td>
373
+ <td>1033</td>
374
+ <td>2.5</td>
375
+ <td>1819</td>
376
+ <td>21.1</td>
377
+ <td>213</td>
378
+ </tr>
379
+ <tr>
380
+ <th rowspan="3" valign="top">A100x1</th>
381
+ <th>deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B</th>
382
+ <td>---</td>
383
+ <td>0.6</td>
384
+ <td>3359</td>
385
+ <td>1.2</td>
386
+ <td>1654</td>
387
+ <td>0.6</td>
388
+ <td>3286</td>
389
+ <td>0.6</td>
390
+ <td>3241</td>
391
+ <td>4.7</td>
392
+ <td>424</td>
393
+ <td>4.9</td>
394
+ <td>411</td>
395
+ <td>2.6</td>
396
+ <td>778</td>
397
+ <td>21.1</td>
398
+ <td>95</td>
399
+ </tr>
400
+ <tr>
401
+ <th>neuralmagic/DeepSeek-R1-Distill-Qwen-1.5B-quantized.w8a8</th>
402
+ <td>1.05</td>
403
+ <td>0.6</td>
404
+ <td>3531</td>
405
+ <td>1.1</td>
406
+ <td>1807</td>
407
+ <td>0.6</td>
408
+ <td>3427</td>
409
+ <td>0.6</td>
410
+ <td>3480</td>
411
+ <td>4.5</td>
412
+ <td>448</td>
413
+ <td>4.5</td>
414
+ <td>447</td>
415
+ <td>2.4</td>
416
+ <td>842</td>
417
+ <td>23.5</td>
418
+ <td>86</td>
419
+ </tr>
420
+ <tr>
421
+ <th>neuralmagic/DeepSeek-R1-Distill-Qwen-1.5B-quantized.w4a16</th>
422
+ <td>1.03</td>
423
+ <td>0.6</td>
424
+ <td>3469</td>
425
+ <td>1.1</td>
426
+ <td>1751</td>
427
+ <td>0.6</td>
428
+ <td>3403</td>
429
+ <td>0.6</td>
430
+ <td>3407</td>
431
+ <td>4.5</td>
432
+ <td>447</td>
433
+ <td>4.6</td>
434
+ <td>435</td>
435
+ <td>2.5</td>
436
+ <td>815</td>
437
+ <td>23.3</td>
438
+ <td>86</td>
439
+ </tr>
440
+ <tr>
441
+ <th rowspan="3" valign="top">H100x1</th>
442
+ <th>deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B</th>
443
+ <td>---</td>
444
+ <td>0.4</td>
445
+ <td>2604</td>
446
+ <td>0.8</td>
447
+ <td>1299</td>
448
+ <td>0.4</td>
449
+ <td>2543</td>
450
+ <td>0.4</td>
451
+ <td>2551</td>
452
+ <td>3.3</td>
453
+ <td>330</td>
454
+ <td>3.4</td>
455
+ <td>326</td>
456
+ <td>1.8</td>
457
+ <td>612</td>
458
+ <td>14.0</td>
459
+ <td>78</td>
460
+ </tr>
461
+ <tr>
462
+ <th>neuralmagic/DeepSeek-R1-Distill-Qwen-1.5B-FP8-dynamic</th>
463
+ <td>1.04</td>
464
+ <td>0.4</td>
465
+ <td>2694</td>
466
+ <td>0.8</td>
467
+ <td>1364</td>
468
+ <td>0.4</td>
469
+ <td>2670</td>
470
+ <td>0.4</td>
471
+ <td>2639</td>
472
+ <td>3.2</td>
473
+ <td>347</td>
474
+ <td>3.2</td>
475
+ <td>341</td>
476
+ <td>1.6</td>
477
+ <td>673</td>
478
+ <td>14.1</td>
479
+ <td>78</td>
480
+ </tr>
481
+ <tr>
482
+ <th>neuralmagic/DeepSeek-R1-Distill-Qwen-1.5B-quantized.w4a16</th>
483
+ <td>0.84</td>
484
+ <td>0.5</td>
485
+ <td>2111</td>
486
+ <td>1.0</td>
487
+ <td>1065</td>
488
+ <td>0.5</td>
489
+ <td>2068</td>
490
+ <td>0.5</td>
491
+ <td>2119</td>
492
+ <td>4.1</td>
493
+ <td>270</td>
494
+ <td>4.1</td>
495
+ <td>265</td>
496
+ <td>2.1</td>
497
+ <td>530</td>
498
+ <td>15.1</td>
499
+ <td>73</td>
500
+ </tr>
501
+ </tbody>
502
+ </table>
503
+
504
+ **Use case profiles: prompt tokens / generation tokens
505
+
506
+ **QPD: Queries per dollar, based on on-demand cost at [Lambda Labs](https://lambdalabs.com/service/gpu-cloud) (observed on 2/18/2025).