lm-sys/FastChat

Issues with VLLM Integration Speedup

開放

#2,362 建立於 2023年9月5日

 (4 則留言) (0 個反應) (0 位負責人)Python (4,736 個分叉)batch import
good first issue

倉庫指標

星標
 (38,959 顆星)
PR 合併指標
 (PR 指標待抓取)

描述

Hello,

I've been trying to work with the [vLLM integration] and I'm facing some performance discrepancies. According to the documentation, I should achieve a significant speedup, but in my tests, I'm seeing different results:

Directly running with FastChat: 16 t/s Using the VLLM integration: 25 t/s (only 1.5x speedup) VLLM offline inference: 90 t/s (expected 6x speedup) I'm running Vicuna33b on a gin H100 gpu. Has anyone experienced this before? Are there any additional configurations or tweaks I might be missing to get the desired speedup?

Thanks in advance for any guidance or advice!

貢獻者指南