What problem does it solve?
This Skill solves the challenge of efficiently deploying production LLM APIs, optimizing inference latency/throughput, and serving models with limited GPU memory.
Core Features & Use Cases
- High Throughput: Achieves 24x higher throughput than standard transformers.
- OpenAI API Compatibility: Supports OpenAI-compatible endpoints.
- Quantization: Supports quantization (GPTQ/AWQ/FP8) to save GPU memory.
- Tensor Parallelism: Supports tensor parallelism for large models.
- Use Cases: Ideal for multi-user applications (chatbots, assistants) that require low latency and high throughput, serving OpenAI-compatible endpoints, and large models with limited GPU memory.
Quick Start
Install the vLLM package using pip install vllm. Then, you can use it for basic offline inference as follows:
from vllm import LLM, SamplingParams
llm = LLM(model="meta-llama/Llama-3-8B-Instruct")
sampling = SamplingParams(temperature=0.7, max_tokens=256)
outputs = llm.generate(["Explain quantum computing"], sampling)
print(outputs[0].outputs[0].text)