191 lines
6.4 KiB
Markdown
191 lines
6.4 KiB
Markdown
(offline-inference)=
|
|
|
|
# Offline Inference
|
|
|
|
You can run vLLM in your own code on a list of prompts.
|
|
|
|
The offline API is based on the {class}`~vllm.LLM` class.
|
|
To initialize the vLLM engine, create a new instance of `LLM` and specify the model to run.
|
|
|
|
For example, the following code downloads the [`facebook/opt-125m`](https://huggingface.co/facebook/opt-125m) model from HuggingFace
|
|
and runs it in vLLM using the default configuration.
|
|
|
|
```python
|
|
from vllm import LLM
|
|
|
|
llm = LLM(model="facebook/opt-125m")
|
|
```
|
|
|
|
After initializing the `LLM` instance, you can perform model inference using various APIs.
|
|
The available APIs depend on the type of model that is being run:
|
|
|
|
- [Generative models](#generative-models) output logprobs which are sampled from to obtain the final output text.
|
|
- [Pooling models](#pooling-models) output their hidden states directly.
|
|
|
|
Please refer to the above pages for more details about each API.
|
|
|
|
:::{seealso}
|
|
[API Reference](/api/offline_inference/index)
|
|
:::
|
|
|
|
## Configuration Options
|
|
|
|
This section lists the most common options for running the vLLM engine.
|
|
For a full list, refer to the [Engine Arguments](#engine-args) page.
|
|
|
|
(model-resolution)=
|
|
|
|
### Model resolution
|
|
|
|
vLLM loads HuggingFace-compatible models by inspecting the `architectures` field in `config.json` of the model repository
|
|
and finding the corresponding implementation that is registered to vLLM.
|
|
Nevertheless, our model resolution may fail for the following reasons:
|
|
|
|
- The `config.json` of the model repository lacks the `architectures` field.
|
|
- Unofficial repositories refer to a model using alternative names which are not recorded in vLLM.
|
|
- The same architecture name is used for multiple models, creating ambiguity as to which model should be loaded.
|
|
|
|
To fix this, explicitly specify the model architecture by passing `config.json` overrides to the `hf_overrides` option.
|
|
For example:
|
|
|
|
```python
|
|
from vllm import LLM
|
|
|
|
model = LLM(
|
|
model="cerebras/Cerebras-GPT-1.3B",
|
|
hf_overrides={"architectures": ["GPT2LMHeadModel"]}, # GPT-2
|
|
)
|
|
```
|
|
|
|
Our [list of supported models](#supported-models) shows the model architectures that are recognized by vLLM.
|
|
|
|
(reducing-memory-usage)=
|
|
|
|
### Reducing memory usage
|
|
|
|
Large models might cause your machine to run out of memory (OOM). Here are some options that help alleviate this problem.
|
|
|
|
#### Tensor Parallelism (TP)
|
|
|
|
Tensor parallelism (`tensor_parallel_size` option) can be used to split the model across multiple GPUs.
|
|
|
|
The following code splits the model across 2 GPUs.
|
|
|
|
```python
|
|
llm = LLM(model="ibm-granite/granite-3.1-8b-instruct",
|
|
tensor_parallel_size=2)
|
|
```
|
|
|
|
:::{important}
|
|
To ensure that vLLM initializes CUDA correctly, you should avoid calling related functions (e.g. {func}`torch.cuda.set_device`)
|
|
before initializing vLLM. Otherwise, you may run into an error like `RuntimeError: Cannot re-initialize CUDA in forked subprocess`.
|
|
|
|
To control which devices are used, please instead set the `CUDA_VISIBLE_DEVICES` environment variable.
|
|
:::
|
|
|
|
:::{note}
|
|
With tensor parallelism enabled, each process will read the whole model and split it into chunks, which makes the disk reading time even longer (proportional to the size of tensor parallelism).
|
|
|
|
You can convert the model checkpoint to a sharded checkpoint using <gh-file:examples/offline_inference/save_sharded_state.py>. The conversion process might take some time, but later you can load the sharded checkpoint much faster. The model loading time should remain constant regardless of the size of tensor parallelism.
|
|
:::
|
|
|
|
#### Quantization
|
|
|
|
Quantized models take less memory at the cost of lower precision.
|
|
|
|
Statically quantized models can be downloaded from HF Hub (some popular ones are available at [Neural Magic](https://huggingface.co/neuralmagic))
|
|
and used directly without extra configuration.
|
|
|
|
Dynamic quantization is also supported via the `quantization` option -- see [here](#quantization-index) for more details.
|
|
|
|
#### Context length and batch size
|
|
|
|
You can further reduce memory usage by limiting the context length of the model (`max_model_len` option)
|
|
and the maximum batch size (`max_num_seqs` option).
|
|
|
|
```python
|
|
from vllm import LLM
|
|
|
|
llm = LLM(model="adept/fuyu-8b",
|
|
max_model_len=2048,
|
|
max_num_seqs=2)
|
|
```
|
|
|
|
#### Reduce CUDA Graphs
|
|
|
|
By default, we optimize model inference using CUDA graphs which take up extra memory in the GPU.
|
|
|
|
:::{important}
|
|
CUDA graph capture takes up more memory in V1 than in V0.
|
|
:::
|
|
|
|
You can adjust `compilation_config` to achieve a better balance between inference speed and memory usage:
|
|
|
|
```python
|
|
from vllm import LLM
|
|
from vllm.config import CompilationConfig, CompilationLevel
|
|
|
|
llm = LLM(
|
|
model="meta-llama/Llama-3.1-8B-Instruct",
|
|
compilation_config=CompilationConfig(
|
|
level=CompilationLevel.PIECEWISE,
|
|
# By default, it goes up to max_num_seqs
|
|
cudagraph_capture_sizes=[1, 2, 4, 8, 16],
|
|
),
|
|
)
|
|
```
|
|
|
|
You can disable graph capturing completely via the `enforce_eager` flag:
|
|
|
|
```python
|
|
from vllm import LLM
|
|
|
|
llm = LLM(model="meta-llama/Llama-3.1-8B-Instruct",
|
|
enforce_eager=True)
|
|
```
|
|
|
|
#### Adjust cache size
|
|
|
|
If you run out of CPU RAM, try the following options:
|
|
|
|
- (Multi-modal models only) you can set the size of multi-modal input cache using `VLLM_MM_INPUT_CACHE_GIB` environment variable (default 4 GiB).
|
|
- (CPU backend only) you can set the size of KV cache using `VLLM_CPU_KVCACHE_SPACE` environment variable (default 4 GiB).
|
|
|
|
#### Multi-modal input limits
|
|
|
|
You can allow a smaller number of multi-modal items per prompt to reduce the memory footprint of the model:
|
|
|
|
```python
|
|
from vllm import LLM
|
|
|
|
# Accept up to 3 images and 1 video per prompt
|
|
llm = LLM(model="Qwen/Qwen2.5-VL-3B-Instruct",
|
|
limit_mm_per_prompt={"image": 3, "video": 1})
|
|
```
|
|
|
|
You can go a step further and disable unused modalities completely by setting its limit to zero.
|
|
For example, if your application only accepts image input, there is no need to allocate any memory for videos.
|
|
|
|
```python
|
|
from vllm import LLM
|
|
|
|
# Accept any number of images but no videos
|
|
llm = LLM(model="Qwen/Qwen2.5-VL-3B-Instruct",
|
|
limit_mm_per_prompt={"video": 0})
|
|
```
|
|
|
|
You can even run a multi-modal model for text-only inference:
|
|
|
|
```python
|
|
from vllm import LLM
|
|
|
|
# Don't accept images. Just text.
|
|
llm = LLM(model="google/gemma-3-27b-it",
|
|
limit_mm_per_prompt={"image": 0})
|
|
```
|
|
|
|
### Performance optimization and tuning
|
|
|
|
You can potentially improve the performance of vLLM by finetuning various options.
|
|
Please refer to [this guide](#optimization-and-tuning) for more details.
|