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!

贡献者指南