lm-sys / lm-sys/FastChat

Issues with VLLM Integration Speedup

Open
#2,362 4 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

good first issue
Dominant language
Python
Stars
39.5k
Forks
4.8k
PR merge metrics
No merged PRs in 30d

Description

Hello,

I've been trying to work with the [[vLLM integration](https://github.com/lm-sys/FastChat/blob/main/docs/vllm_integration.md)] 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!

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start with docs/vllm_integration.md and compare its documented setup with the reported Vicuna-33B benchmarks on an H100. Reproduce the three inference paths and inspect the configuration differences; done means identifying the cause of the speed discrepancy or documenting the required settings and expected results.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning, performance
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.