lm-sys/FastChat

Issues with VLLM Integration Speedup

Open

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

在 GitHub 查看
 (4 留言) (0 反應) (0 負責人)Python (4,736 fork)batch import
good first issue

倉庫指標

Star
 (38,959 star)
PR 合併指標
 (30 天內沒有已合併 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!

貢獻者指南