mirror of
https://github.com/mudler/LocalAI.git
synced 2026-07-31 02:18:50 -04:00
The pin sat at f384edcd while vllm.cpp main moved a long way. The ABI is unchanged at v10 and both POD structs are field-identical to the pinned commit (verified by diffing vllm_model_params and vllm_sampling_params across the range), so the Go mirror needs no edit and this is a clean bump. What it picks up matters for this backend: - MTP speculative decoding from a GGUF target, gated end to end on GPU. - DFlash speculative decoding with a GGUF draft AND a GGUF target. - NVFP4 GGUF: dequant, plus a native fp4 compute path for dense and full-attention projections. On the 27B that closed a cross-container divergence entirely (the GGUF and safetensors builds of the same quantization run now emit identical tokens) and halved peak RSS. - A real engine fix: the GDN speculative state gather/scatter was mis-striding the widened conv row, so speculation silently corrupted the target's own recurrent state on CPU. Docs corrected accordingly. The section previously told users that mtp and dflash are rejected on a .gguf target and called it a gap in the engine's GGUF loader. That is no longer true, and leaving it would send people to safetensors for no reason. A head-less GGUF is still refused, and the text now says so with the actual cause. The real-library ABI handshake was re-run against a libvllm.so built at the exact pinned commit rather than a stale one: 43 specs pass, reported ABI 10. make lint clean. Signed-off-by: Ettore Di Giacinto <mudler@localai.io> Assisted-by: Claude Code:claude-opus-5 [ClaudeCode]
1236 lines
48 KiB
Markdown
1236 lines
48 KiB
Markdown
|
||
+++
|
||
disableToc = false
|
||
title = "Text Generation (GPT)"
|
||
weight = 10
|
||
url = "/features/text-generation/"
|
||
+++
|
||
|
||
LocalAI supports generating text with GPT with `llama.cpp` and other backends (such as `rwkv.cpp` as ) see also the [Model compatibility]({{%relref "reference/compatibility-table" %}}) for an up-to-date list of the supported model families.
|
||
|
||
Note:
|
||
|
||
- You can also specify the model name as part of the OpenAI token.
|
||
- If only one model is available, the API will use it for all the requests.
|
||
|
||
## API Reference
|
||
|
||
### Chat completions
|
||
|
||
https://platform.openai.com/docs/api-reference/chat
|
||
|
||
For example, to generate a chat completion, you can send a POST request to the `/v1/chat/completions` endpoint with the instruction as the request body:
|
||
|
||
```bash
|
||
curl http://localhost:8080/v1/chat/completions -H "Content-Type: application/json" -d '{
|
||
"model": "ggml-koala-7b-model-q4_0-r2.bin",
|
||
"messages": [{"role": "user", "content": "Say this is a test!"}],
|
||
"temperature": 0.7
|
||
}'
|
||
```
|
||
|
||
Available additional parameters: `top_p`, `top_k`, `max_tokens`
|
||
|
||
Reasoning models return their thinking in the `reasoning` field. When a model reasons and calls a tool in the same turn, see [Interleaved Thinking with Tool Calls]({{%relref "features/interleaved-thinking" %}}).
|
||
|
||
### Edit completions
|
||
|
||
https://platform.openai.com/docs/api-reference/edits
|
||
|
||
To generate an edit completion you can send a POST request to the `/v1/edits` endpoint with the instruction as the request body:
|
||
|
||
```bash
|
||
curl http://localhost:8080/v1/edits -H "Content-Type: application/json" -d '{
|
||
"model": "ggml-koala-7b-model-q4_0-r2.bin",
|
||
"instruction": "rephrase",
|
||
"input": "Black cat jumped out of the window",
|
||
"temperature": 0.7
|
||
}'
|
||
```
|
||
|
||
Available additional parameters: `top_p`, `top_k`, `max_tokens`.
|
||
|
||
### Completions
|
||
|
||
https://platform.openai.com/docs/api-reference/completions
|
||
|
||
To generate a completion, you can send a POST request to the `/v1/completions` endpoint with the instruction as per the request body:
|
||
|
||
```bash
|
||
curl http://localhost:8080/v1/completions -H "Content-Type: application/json" -d '{
|
||
"model": "ggml-koala-7b-model-q4_0-r2.bin",
|
||
"prompt": "A long time ago in a galaxy far, far away",
|
||
"temperature": 0.7
|
||
}'
|
||
```
|
||
|
||
Available additional parameters: `top_p`, `top_k`, `max_tokens`
|
||
|
||
### List models
|
||
|
||
You can list all the models available with:
|
||
|
||
```bash
|
||
curl http://localhost:8080/v1/models
|
||
```
|
||
|
||
### Anthropic Messages API
|
||
|
||
LocalAI supports the Anthropic Messages API, which is compatible with Claude clients. This endpoint provides a structured way to send messages and receive responses, with support for tools, streaming, and multimodal content.
|
||
|
||
**Endpoint:** `POST /v1/messages` or `POST /messages`
|
||
|
||
**Reference:** https://docs.anthropic.com/claude/reference/messages_post
|
||
|
||
#### Basic Usage
|
||
|
||
```bash
|
||
curl http://localhost:8080/v1/messages \
|
||
-H "Content-Type: application/json" \
|
||
-H "anthropic-version: 2023-06-01" \
|
||
-d '{
|
||
"model": "ggml-koala-7b-model-q4_0-r2.bin",
|
||
"max_tokens": 1024,
|
||
"messages": [
|
||
{"role": "user", "content": "Say this is a test!"}
|
||
]
|
||
}'
|
||
```
|
||
|
||
#### Request Parameters
|
||
|
||
| Parameter | Type | Required | Description |
|
||
|-----------|------|----------|-------------|
|
||
| `model` | string | Yes | The model identifier |
|
||
| `messages` | array | Yes | Array of message objects with `role` and `content` |
|
||
| `max_tokens` | integer | Yes | Maximum number of tokens to generate (must be > 0) |
|
||
| `system` | string | No | System message to set the assistant's behavior |
|
||
| `temperature` | float | No | Sampling temperature (0.0 to 1.0) |
|
||
| `top_p` | float | No | Nucleus sampling parameter |
|
||
| `top_k` | integer | No | Top-k sampling parameter |
|
||
| `stop_sequences` | array | No | Array of strings that will stop generation |
|
||
| `stream` | boolean | No | Enable streaming responses |
|
||
| `tools` | array | No | Array of tool definitions for function calling |
|
||
| `tool_choice` | string/object | No | Tool choice strategy: "auto", "any", "none", or specific tool |
|
||
| `metadata` | object | No | Per-request metadata passed to the backend (e.g., `{"enable_thinking": "true"}`) |
|
||
|
||
#### Message Format
|
||
|
||
Messages can contain text or structured content blocks:
|
||
|
||
```bash
|
||
curl http://localhost:8080/v1/messages \
|
||
-H "Content-Type: application/json" \
|
||
-d '{
|
||
"model": "ggml-koala-7b-model-q4_0-r2.bin",
|
||
"max_tokens": 1024,
|
||
"messages": [
|
||
{
|
||
"role": "user",
|
||
"content": [
|
||
{
|
||
"type": "text",
|
||
"text": "What is in this image?"
|
||
},
|
||
{
|
||
"type": "image",
|
||
"source": {
|
||
"type": "base64",
|
||
"media_type": "image/jpeg",
|
||
"data": "base64_encoded_image_data"
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
}'
|
||
```
|
||
|
||
#### Tool Calling
|
||
|
||
The Anthropic API supports function calling through tools:
|
||
|
||
```bash
|
||
curl http://localhost:8080/v1/messages \
|
||
-H "Content-Type: application/json" \
|
||
-d '{
|
||
"model": "ggml-koala-7b-model-q4_0-r2.bin",
|
||
"max_tokens": 1024,
|
||
"tools": [
|
||
{
|
||
"name": "get_weather",
|
||
"description": "Get the current weather",
|
||
"input_schema": {
|
||
"type": "object",
|
||
"properties": {
|
||
"location": {
|
||
"type": "string",
|
||
"description": "The city and state"
|
||
}
|
||
},
|
||
"required": ["location"]
|
||
}
|
||
}
|
||
],
|
||
"tool_choice": "auto",
|
||
"messages": [
|
||
{"role": "user", "content": "What is the weather in San Francisco?"}
|
||
]
|
||
}'
|
||
```
|
||
|
||
#### Streaming
|
||
|
||
Enable streaming responses by setting `stream: true`:
|
||
|
||
```bash
|
||
curl http://localhost:8080/v1/messages \
|
||
-H "Content-Type: application/json" \
|
||
-d '{
|
||
"model": "ggml-koala-7b-model-q4_0-r2.bin",
|
||
"max_tokens": 1024,
|
||
"stream": true,
|
||
"messages": [
|
||
{"role": "user", "content": "Tell me a story"}
|
||
]
|
||
}'
|
||
```
|
||
|
||
Streaming responses use Server-Sent Events (SSE) format with event types: `message_start`, `content_block_start`, `content_block_delta`, `content_block_stop`, `message_delta`, and `message_stop`.
|
||
|
||
#### Response Format
|
||
|
||
```json
|
||
{
|
||
"id": "msg_abc123",
|
||
"type": "message",
|
||
"role": "assistant",
|
||
"content": [
|
||
{
|
||
"type": "text",
|
||
"text": "This is a test!"
|
||
}
|
||
],
|
||
"model": "ggml-koala-7b-model-q4_0-r2.bin",
|
||
"stop_reason": "end_turn",
|
||
"usage": {
|
||
"input_tokens": 10,
|
||
"output_tokens": 5
|
||
}
|
||
}
|
||
```
|
||
|
||
### Open Responses API
|
||
|
||
LocalAI supports the Open Responses API specification, which provides a standardized interface for AI model interactions with support for background processing, streaming, tool calling, and advanced features like reasoning.
|
||
|
||
**Endpoint:** `POST /v1/responses` or `POST /responses`
|
||
|
||
**Reference:** https://www.openresponses.org/specification
|
||
|
||
#### Basic Usage
|
||
|
||
```bash
|
||
curl http://localhost:8080/v1/responses \
|
||
-H "Content-Type: application/json" \
|
||
-d '{
|
||
"model": "ggml-koala-7b-model-q4_0-r2.bin",
|
||
"input": "Say this is a test!",
|
||
"max_output_tokens": 1024
|
||
}'
|
||
```
|
||
|
||
#### Request Parameters
|
||
|
||
| Parameter | Type | Required | Description |
|
||
|-----------|------|----------|-------------|
|
||
| `model` | string | Yes | The model identifier |
|
||
| `input` | string/array | Yes | Input text or array of input items |
|
||
| `max_output_tokens` | integer | No | Maximum number of tokens to generate |
|
||
| `temperature` | float | No | Sampling temperature |
|
||
| `top_p` | float | No | Nucleus sampling parameter |
|
||
| `instructions` | string | No | System instructions |
|
||
| `tools` | array | No | Array of tool definitions |
|
||
| `tool_choice` | string/object | No | Tool choice: "auto", "required", "none", or specific tool |
|
||
| `stream` | boolean | No | Enable streaming responses |
|
||
| `background` | boolean | No | Run request in background (returns immediately) |
|
||
| `store` | boolean | No | Whether to store the response |
|
||
| `reasoning` | object | No | Reasoning configuration with `effort` and `summary` |
|
||
| `parallel_tool_calls` | boolean | No | Allow parallel tool calls |
|
||
| `max_tool_calls` | integer | No | Maximum number of tool calls |
|
||
| `presence_penalty` | float | No | Presence penalty (-2.0 to 2.0) |
|
||
| `frequency_penalty` | float | No | Frequency penalty (-2.0 to 2.0) |
|
||
| `top_logprobs` | integer | No | Number of top logprobs to return |
|
||
| `truncation` | string | No | Truncation mode: "auto" or "disabled" |
|
||
| `text_format` | object | No | Text format configuration |
|
||
| `metadata` | object | No | Custom metadata |
|
||
|
||
#### Input Format
|
||
|
||
Input can be a simple string or an array of structured items:
|
||
|
||
```bash
|
||
curl http://localhost:8080/v1/responses \
|
||
-H "Content-Type: application/json" \
|
||
-d '{
|
||
"model": "ggml-koala-7b-model-q4_0-r2.bin",
|
||
"input": [
|
||
{
|
||
"type": "message",
|
||
"role": "user",
|
||
"content": "What is the weather?"
|
||
}
|
||
],
|
||
"max_output_tokens": 1024
|
||
}'
|
||
```
|
||
|
||
#### Background Processing
|
||
|
||
Run requests in the background for long-running tasks:
|
||
|
||
```bash
|
||
curl http://localhost:8080/v1/responses \
|
||
-H "Content-Type: application/json" \
|
||
-d '{
|
||
"model": "ggml-koala-7b-model-q4_0-r2.bin",
|
||
"input": "Generate a long story",
|
||
"max_output_tokens": 4096,
|
||
"background": true
|
||
}'
|
||
```
|
||
|
||
The response will include a response ID that can be used to poll for completion:
|
||
|
||
```json
|
||
{
|
||
"id": "resp_abc123",
|
||
"object": "response",
|
||
"status": "in_progress",
|
||
"created_at": 1234567890
|
||
}
|
||
```
|
||
|
||
#### Retrieving Background Responses
|
||
|
||
Use the GET endpoint to retrieve background responses:
|
||
|
||
```bash
|
||
# Get response by ID
|
||
curl http://localhost:8080/v1/responses/resp_abc123
|
||
|
||
# Resume streaming with query parameters
|
||
curl "http://localhost:8080/v1/responses/resp_abc123?stream=true&starting_after=10"
|
||
```
|
||
|
||
#### Canceling Background Responses
|
||
|
||
Cancel a background response that's still in progress:
|
||
|
||
```bash
|
||
curl -X POST http://localhost:8080/v1/responses/resp_abc123/cancel
|
||
```
|
||
|
||
#### Multiple Replicas (Distributed Mode)
|
||
|
||
In distributed mode LocalAI replicates response metadata across frontend
|
||
replicas, so retrieval, `previous_response_id` chaining and cancellation work
|
||
regardless of which replica the load balancer picks:
|
||
|
||
- `GET /v1/responses/{id}` returns the response from any replica.
|
||
- `POST /v1/responses/{id}/cancel` is delegated over NATS to the replica that is
|
||
actually generating, so generation really stops. If that replica is gone, the
|
||
response is reported as `cancelled` without blocking.
|
||
- **Streaming resume (`?stream=true`) is served only by the replica that created
|
||
the response.** The event buffer lives in that process's memory and is not
|
||
replicated. A resume request that reaches another replica returns HTTP 409
|
||
naming the owning replica instead of silently returning a truncated stream.
|
||
Poll the response instead, or route resume requests with session affinity.
|
||
|
||
#### Tool Calling
|
||
|
||
Open Responses API supports function calling with tools:
|
||
|
||
```bash
|
||
curl http://localhost:8080/v1/responses \
|
||
-H "Content-Type: application/json" \
|
||
-d '{
|
||
"model": "ggml-koala-7b-model-q4_0-r2.bin",
|
||
"input": "What is the weather in San Francisco?",
|
||
"tools": [
|
||
{
|
||
"type": "function",
|
||
"name": "get_weather",
|
||
"description": "Get the current weather",
|
||
"parameters": {
|
||
"type": "object",
|
||
"properties": {
|
||
"location": {
|
||
"type": "string",
|
||
"description": "The city and state"
|
||
}
|
||
},
|
||
"required": ["location"]
|
||
}
|
||
}
|
||
],
|
||
"tool_choice": "auto",
|
||
"max_output_tokens": 1024
|
||
}'
|
||
```
|
||
|
||
#### Reasoning Configuration
|
||
|
||
Configure reasoning effort and summary style:
|
||
|
||
```bash
|
||
curl http://localhost:8080/v1/responses \
|
||
-H "Content-Type: application/json" \
|
||
-d '{
|
||
"model": "ggml-koala-7b-model-q4_0-r2.bin",
|
||
"input": "Solve this complex problem step by step",
|
||
"reasoning": {
|
||
"effort": "high",
|
||
"summary": "detailed"
|
||
},
|
||
"max_output_tokens": 2048
|
||
}'
|
||
```
|
||
|
||
#### Response Format
|
||
|
||
```json
|
||
{
|
||
"id": "resp_abc123",
|
||
"object": "response",
|
||
"created_at": 1234567890,
|
||
"completed_at": 1234567895,
|
||
"status": "completed",
|
||
"model": "ggml-koala-7b-model-q4_0-r2.bin",
|
||
"output": [
|
||
{
|
||
"type": "message",
|
||
"id": "msg_001",
|
||
"role": "assistant",
|
||
"content": [
|
||
{
|
||
"type": "output_text",
|
||
"text": "This is a test!",
|
||
"annotations": [],
|
||
"logprobs": []
|
||
}
|
||
],
|
||
"status": "completed"
|
||
}
|
||
],
|
||
"error": null,
|
||
"incomplete_details": null,
|
||
"temperature": 0.7,
|
||
"top_p": 1.0,
|
||
"presence_penalty": 0.0,
|
||
"frequency_penalty": 0.0,
|
||
"usage": {
|
||
"input_tokens": 10,
|
||
"output_tokens": 5,
|
||
"total_tokens": 15,
|
||
"input_tokens_details": {
|
||
"cached_tokens": 0
|
||
},
|
||
"output_tokens_details": {
|
||
"reasoning_tokens": 0
|
||
}
|
||
}
|
||
}
|
||
```
|
||
|
||
## Backends
|
||
|
||
### RWKV
|
||
|
||
RWKV support is available through llama.cpp (see below)
|
||
|
||
### llama.cpp
|
||
|
||
[llama.cpp](https://github.com/ggerganov/llama.cpp) is a popular port of Facebook's LLaMA model in C/C++.
|
||
|
||
{{% notice note %}}
|
||
|
||
The `ggml` file format has been deprecated. If you are using `ggml` models and you are configuring your model with a YAML file, specify, use a LocalAI version older than v2.25.0. For `gguf` models, use the `llama` backend. The go backend is deprecated as well but still available as `go-llama`.
|
||
|
||
{{% /notice %}}
|
||
|
||
#### Features
|
||
|
||
The `llama.cpp` model supports the following features:
|
||
- [📖 Text generation (GPT)]({{%relref "features/text-generation" %}})
|
||
- [🧠 Embeddings]({{%relref "features/embeddings" %}})
|
||
- [🔥 OpenAI functions]({{%relref "features/openai-functions" %}})
|
||
- [✍️ Constrained grammars]({{%relref "features/constrained_grammars" %}})
|
||
|
||
#### Setup
|
||
|
||
LocalAI supports `llama.cpp` models out of the box. You can use the `llama.cpp` model in the same way as any other model.
|
||
|
||
##### Manual setup
|
||
|
||
It is sufficient to copy the `ggml` or `gguf` model files in the `models` folder. You can refer to the model in the `model` parameter in the API calls.
|
||
|
||
[You can optionally create an associated YAML]({{%relref "advanced" %}}) model config file to tune the model's parameters or apply a template to the prompt.
|
||
|
||
Prompt templates are useful for models that are fine-tuned towards a specific prompt.
|
||
|
||
##### Automatic setup
|
||
|
||
LocalAI supports model galleries which are indexes of models. For instance, the huggingface gallery contains a large curated index of models from the huggingface model hub for `ggml` or `gguf` models.
|
||
|
||
For instance, if you have the galleries enabled and LocalAI already running, you can just start chatting with models in huggingface by running:
|
||
|
||
```bash
|
||
curl http://localhost:8080/v1/chat/completions -H "Content-Type: application/json" -d '{
|
||
"model": "TheBloke/WizardLM-13B-V1.2-GGML/wizardlm-13b-v1.2.ggmlv3.q2_K.bin",
|
||
"messages": [{"role": "user", "content": "Say this is a test!"}],
|
||
"temperature": 0.1
|
||
}'
|
||
```
|
||
|
||
LocalAI will automatically download and configure the model in the `model` directory.
|
||
|
||
Models can be also preloaded or downloaded on demand. To learn about model galleries, check out the [model gallery documentation]({{%relref "features/model-gallery" %}}).
|
||
|
||
#### YAML configuration
|
||
|
||
To use the `llama.cpp` backend, specify `llama-cpp` as the backend in the YAML file:
|
||
|
||
```yaml
|
||
name: llama
|
||
backend: llama-cpp
|
||
parameters:
|
||
# Relative to the models path
|
||
model: file.gguf
|
||
```
|
||
|
||
#### Backend Options
|
||
|
||
The `llama.cpp` backend supports additional configuration options that can be specified in the `options` field of your model YAML configuration. These options allow fine-tuning of the backend behavior:
|
||
|
||
| Option | Type | Description | Example |
|
||
|--------|------|-------------|---------|
|
||
| `use_jinja` or `jinja` | boolean | Enable Jinja2 template processing for chat templates. When enabled, the backend uses Jinja2-based chat templates from the model for formatting messages. | `use_jinja:true` |
|
||
| `context_shift` | boolean | Enable context shifting, which allows the model to dynamically adjust context window usage. | `context_shift:true` |
|
||
| `cache_ram` | integer | Size budget in MiB for the **server-side prompt cache** (a host-RAM store of idle slot KV states that's reloaded on a prompt-prefix hit, see [upstream PR #16391](https://github.com/ggml-org/llama.cpp/pull/16391)). Default: `-1` (no limit). `0` disables the prompt cache entirely. Together with `kv_unified` and `cache_idle_slots` this is what makes a repeated system prompt skip prefill on subsequent calls. | `cache_ram:4096` |
|
||
| `parallel` or `n_parallel` | integer | Enable parallel request processing. When set to a value greater than 1, enables continuous batching for handling multiple requests concurrently. | `parallel:4` |
|
||
| `grpc_servers` or `rpc_servers` | string | Comma-separated list of gRPC server addresses for distributed inference. Allows distributing workload across multiple llama.cpp workers. | `grpc_servers:localhost:50051,localhost:50052` |
|
||
| `fit_params` or `fit` | boolean | Enable auto-adjustment of model/context parameters to fit available device memory. Default: `true`. | `fit_params:true` |
|
||
| `fit_params_target` or `fit_target` | integer | Target margin per device in MiB when using fit_params. Default: `1024` (1GB). | `fit_target:2048` |
|
||
| `fit_params_min_ctx` or `fit_ctx` | integer | Minimum context size that can be set by fit_params. Default: `4096`. | `fit_ctx:2048` |
|
||
| `n_cache_reuse` or `cache_reuse` | integer | Minimum chunk size to attempt reusing from the cache via KV shifting. Default: `0` (disabled). | `cache_reuse:256` |
|
||
| `slot_prompt_similarity` or `sps` | float | How much the prompt of a request must match the prompt of a slot to use that slot. Default: `0.1`. Set to `0` to disable. | `sps:0.5` |
|
||
| `swa_full` | boolean | Use full-size SWA (Sliding Window Attention) cache. Default: `false`. | `swa_full:true` |
|
||
| `cont_batching` or `continuous_batching` | boolean | Enable continuous batching for handling multiple sequences. Default: `true`. | `cont_batching:true` |
|
||
| `check_tensors` | boolean | Validate tensor data for invalid values during model loading. Default: `false`. | `check_tensors:true` |
|
||
| `warmup` | boolean | Enable warmup run after model loading. Default: `true`. | `warmup:false` |
|
||
| `no_op_offload` | boolean | Disable offloading host tensor operations to device. Default: `false`. | `no_op_offload:true` |
|
||
| `device` or `devices` | string | Select the llama.cpp backend devices to use. Repeat the option or pass a comma-separated list; unlisted devices are excluded. Use the names reported by `llama-server --list-devices` / `--list-devices`. | `devices:CUDA1,CUDA2,CUDA3` |
|
||
| `kv_unified` or `unified_kv` | boolean | Use a single unified KV buffer shared across all sequences. Default: `true` (LocalAI override; upstream defaults to `false` but auto-enables it when slot count is auto). **Required for `cache_idle_slots` and scoring**: without it the server force-disables idle-slot saving at init, and score-enabled models are rejected at load time. | `kv_unified:false` |
|
||
| `cache_idle_slots` or `idle_slots_cache` | boolean | On a new task, save the previous slot's KV state into the prompt cache (and clear the slot) so a later request with the same prefix can warm-load it. Default: `true`. Auto-disabled by the server if `kv_unified=false` or `cache_ram=0`. | `cache_idle_slots:false` |
|
||
| `n_ctx_checkpoints` or `ctx_checkpoints` | integer | Maximum number of context checkpoints per slot (used for partial-prefix recovery, e.g. SWA). Default: `32`. | `ctx_checkpoints:16` |
|
||
| `checkpoint_min_step` or `checkpoint_min_spacing` (aliases: `checkpoint_every_nt`, `checkpoint_every_n_tokens`) | integer | Minimum spacing in tokens between context checkpoints. `0` disables the minimum-spacing gate. Default: `256`. (Renamed upstream from `checkpoint_every_nt`; semantics shifted from a fixed cadence to a minimum spacing.) | `checkpoint_min_step:1024` |
|
||
| `split_mode` or `sm` | string | How to split the model across multiple GPUs: `none` (single GPU only), `layer` (default - split layers and KV across GPUs), `row` (split rows across GPUs), `tensor` (experimental tensor parallelism, requires `flash_attention: true`, manually set `context_size`, and a llama.cpp build that includes [#19378](https://github.com/ggml-org/llama.cpp/pull/19378); it historically also required KV-cache quantization to be disabled, but [#23792](https://github.com/ggml-org/llama.cpp/pull/23792) lifts that restriction so `cache_type_k`/`cache_type_v` quantization can be combined with tensor parallelism on builds that include it). | `split_mode:tensor` |
|
||
|
||
**Example configuration with options:**
|
||
|
||
```yaml
|
||
name: llama-model
|
||
backend: llama
|
||
parameters:
|
||
model: model.gguf
|
||
options:
|
||
- use_jinja:true
|
||
- context_shift:true
|
||
- cache_ram:4096
|
||
- parallel:2
|
||
- devices:CUDA1,CUDA2,CUDA3
|
||
- fit_params:true
|
||
- fit_target:1024
|
||
- slot_prompt_similarity:0.5
|
||
```
|
||
|
||
**Note:** The `parallel` option can also be set via the `LLAMACPP_PARALLEL` environment variable, and `grpc_servers` can be set via the `LLAMACPP_GRPC_SERVERS` environment variable. Options specified in the YAML file take precedence over environment variables.
|
||
|
||
##### Hardware auto-tuning (and how to override it)
|
||
|
||
On a detected GPU, LocalAI fills a few performance-relevant defaults the model config leaves unset - a larger physical batch on NVIDIA Blackwell, and a VRAM-scaled `parallel` slot count for concurrent serving. Both are gated on **per-device** VRAM at the model's context: when a large context already fills a single card (e.g. a 27B model with a 200k context across 2×16 GiB), the batch boost and the extra parallel slots are suppressed so they can't tip the tighter GPU into CUDA out-of-memory.
|
||
|
||
Anything you set explicitly in the model YAML always wins, so to pin a value just set it (e.g. `batch: 512` or `options: ["parallel:1"]`). The effective values are logged at `INFO` when a model loads (`effective runtime tuning …`). To turn the hardware auto-tuning off entirely and run llama.cpp's stock behavior, set:
|
||
|
||
```
|
||
LOCALAI_DISABLE_HARDWARE_DEFAULTS=true
|
||
```
|
||
|
||
##### Server-side prompt cache (repeated system prompts)
|
||
|
||
Agents, coding assistants, and Anthropic/OpenAI-compatible CLIs typically resend the same large system prompt on every turn. The llama.cpp server can short-circuit prefill for the matching prefix by stashing idle slot KV states in host RAM and reloading them on a hit. Three settings interact:
|
||
|
||
| Setting | Default | Role |
|
||
|---|---|---|
|
||
| `cache_ram:N` | `-1` (no limit) | Allocates the host-side prompt cache. `0` disables it. |
|
||
| `kv_unified:true` | `true` | Single unified KV buffer (**prerequisite** for idle-slot saving). |
|
||
| `cache_idle_slots:true` | `true` | Persists the idle slot's KV into the prompt cache on task switch. |
|
||
|
||
All three are on by default since LocalAI v4.3, so the prompt cache works out of the box for the common single-slot setup. If you're on an older release, or you've explicitly disabled one of them, add the following to recover the behaviour:
|
||
|
||
```yaml
|
||
options:
|
||
- cache_ram:4096 # or -1 for no limit
|
||
- kv_unified:true
|
||
- cache_idle_slots:true
|
||
```
|
||
|
||
Set `cache_ram:0` to opt out of the prompt cache entirely (saves host RAM at the cost of re-prefilling repeated prompts).
|
||
|
||
#### Reference
|
||
|
||
- [llama](https://github.com/ggerganov/llama.cpp)
|
||
|
||
|
||
### ik_llama.cpp
|
||
|
||
[ik_llama.cpp](https://github.com/ikawrakow/ik_llama.cpp) is a hard fork of `llama.cpp` by Iwan Kawrakow that focuses on superior CPU and hybrid GPU/CPU performance. It ships additional quantization types (IQK quants), custom quantization mixes, Multi-head Latent Attention (MLA) for DeepSeek models, and fine-grained tensor offload controls - particularly useful for running very large models on commodity CPU hardware.
|
||
|
||
{{% notice note %}}
|
||
|
||
The `ik-llama-cpp` backend requires a CPU with **AVX2** support. The IQK kernels are not compatible with older CPUs.
|
||
|
||
{{% /notice %}}
|
||
|
||
#### Features
|
||
|
||
The `ik-llama-cpp` backend supports the following features:
|
||
- [📖 Text generation (GPT)]({{%relref "features/text-generation" %}})
|
||
- [🧠 Embeddings]({{%relref "features/embeddings" %}})
|
||
- IQK quantization types for better CPU inference performance
|
||
- Multimodal models (via clip/llava)
|
||
|
||
#### Setup
|
||
|
||
The backend is distributed as a separate container image and can be installed from the LocalAI backend gallery, or specified directly in a model configuration. GGUF models loaded with this backend benefit from ik_llama.cpp's optimized CPU kernels - especially useful for MoE models and large quantized models that would otherwise be GPU-bound.
|
||
|
||
#### YAML configuration
|
||
|
||
To use the `ik-llama-cpp` backend, specify it as the backend in the YAML file:
|
||
|
||
```yaml
|
||
name: my-model
|
||
backend: ik-llama-cpp
|
||
parameters:
|
||
# Relative to the models path
|
||
model: file.gguf
|
||
```
|
||
|
||
The aliases `ik-llama` and `ik_llama` are also accepted.
|
||
|
||
#### Reference
|
||
|
||
- [ik_llama.cpp](https://github.com/ikawrakow/ik_llama.cpp)
|
||
|
||
|
||
### turboquant (llama.cpp fork with TurboQuant KV-cache)
|
||
|
||
[llama-cpp-turboquant](https://github.com/TheTom/llama-cpp-turboquant) is a `llama.cpp` fork that adds the **TurboQuant KV-cache** quantization scheme. It reuses the upstream `llama.cpp` codebase and ships as a drop-in alternative backend inside LocalAI, sharing the same gRPC server sources as the stock `llama-cpp` backend - so any GGUF model that runs on `llama-cpp` also runs on `turboquant`.
|
||
|
||
You would pick `turboquant` when you want **smaller KV-cache memory pressure** (longer contexts on the same VRAM) or to experiment with the fork's quantized KV representations on top of the standard `cache_type_k` / `cache_type_v` knobs already supported by upstream `llama.cpp`.
|
||
|
||
#### Features
|
||
|
||
- Drop-in GGUF compatibility with upstream `llama.cpp`.
|
||
- TurboQuant KV-cache quantization (see fork README for the current set of accepted `cache_type_k` / `cache_type_v` values).
|
||
- Same feature surface as the `llama-cpp` backend: text generation, embeddings, tool calls, multimodal via mmproj.
|
||
- Available on CPU (AVX/AVX2/AVX512/fallback), NVIDIA CUDA 12/13, AMD ROCm/HIP, Intel SYCL f32/f16, Vulkan, and NVIDIA L4T.
|
||
|
||
#### Setup
|
||
|
||
`turboquant` ships as a separate container image in the LocalAI backend gallery. Install it like any other backend:
|
||
|
||
```bash
|
||
local-ai backends install turboquant
|
||
```
|
||
|
||
Or pick a specific flavor for your hardware (example tags: `cpu-turboquant`, `cuda12-turboquant`, `cuda13-turboquant`, `rocm-turboquant`, `intel-sycl-f16-turboquant`, `vulkan-turboquant`).
|
||
|
||
#### YAML configuration
|
||
|
||
To run a model with `turboquant`, set the backend in your model YAML and optionally pick quantized KV-cache types:
|
||
|
||
```yaml
|
||
name: my-model
|
||
backend: turboquant
|
||
parameters:
|
||
# Relative to the models path
|
||
model: file.gguf
|
||
# Use TurboQuant's own KV-cache quantization schemes. The fork accepts
|
||
# the standard llama.cpp types (f16, f32, q8_0, q4_0, q4_1, q5_0, q5_1)
|
||
# and adds three TurboQuant-specific ones: turbo2, turbo3, turbo4.
|
||
# turbo3 / turbo4 auto-enable flash_attention (required for turbo K/V)
|
||
# and offer progressively more aggressive compression.
|
||
cache_type_k: turbo3
|
||
cache_type_v: turbo3
|
||
context_size: 8192
|
||
```
|
||
|
||
The `cache_type_k` / `cache_type_v` fields map to llama.cpp's `-ctk` / `-ctv` flags. The stock `llama-cpp` backend only accepts the standard llama.cpp types - to use `turbo2` / `turbo3` / `turbo4` you need this `turboquant` backend, which is where the fork's TurboQuant code paths actually take effect. Pick `q8_0` here and you're just running stock llama.cpp KV quantization; pick `turbo*` and you're running TurboQuant.
|
||
|
||
#### Reference
|
||
|
||
- [llama-cpp-turboquant](https://github.com/TheTom/llama-cpp-turboquant)
|
||
- [Tracked branch: `feature/turboquant-kv-cache`](https://github.com/TheTom/llama-cpp-turboquant/tree/feature/turboquant-kv-cache)
|
||
|
||
|
||
### vLLM
|
||
|
||
[vLLM](https://github.com/vllm-project/vllm) is a fast and easy-to-use library for LLM inference.
|
||
|
||
LocalAI has a built-in integration with vLLM, and it can be used to run models. You can check out `vllm` performance [here](https://github.com/vllm-project/vllm#performance).
|
||
|
||
#### Setup
|
||
|
||
Create a YAML file for the model you want to use with `vllm`.
|
||
|
||
To setup a model, you need to just specify the model name in the YAML config file:
|
||
```yaml
|
||
name: vllm
|
||
backend: vllm
|
||
parameters:
|
||
model: "facebook/opt-125m"
|
||
|
||
```
|
||
|
||
The backend will automatically download the required files in order to run the model.
|
||
|
||
|
||
#### Usage
|
||
|
||
Use the `completions` endpoint by specifying the `vllm` backend:
|
||
```
|
||
curl http://localhost:8080/v1/completions -H "Content-Type: application/json" -d '{
|
||
"model": "vllm",
|
||
"prompt": "Hello, my name is",
|
||
"temperature": 0.1, "top_p": 0.1
|
||
}'
|
||
```
|
||
|
||
#### Passing arbitrary vLLM options with `engine_args`
|
||
|
||
A subset of `AsyncEngineArgs` is exposed as typed YAML fields
|
||
(`tensor_parallel_size`, `gpu_memory_utilization`, `quantization`,
|
||
`max_model_len`, `dtype`, `trust_remote_code`, `enforce_eager`, …).
|
||
Anything else can be passed through the generic `engine_args:` map.
|
||
Keys are forwarded verbatim to vLLM's engine; unknown keys fail at load
|
||
time with the closest valid name as a hint. Nested maps materialise
|
||
into vLLM's nested config dataclasses (`SpeculativeConfig`,
|
||
`KVTransferConfig`, `CompilationConfig`, …).
|
||
|
||
Speculative decoding (DFlash, ngram, eagle, deepseek_mtp, …) is
|
||
configured this way:
|
||
|
||
```yaml
|
||
name: qwen3.5-4b-dflash
|
||
backend: vllm
|
||
parameters:
|
||
model: Qwen/Qwen3.5-4B
|
||
context_size: 8192
|
||
max_model_len: 8192
|
||
trust_remote_code: true
|
||
quantization: fp8
|
||
template:
|
||
use_tokenizer_template: true
|
||
engine_args:
|
||
speculative_config:
|
||
method: dflash
|
||
model: z-lab/Qwen3.5-4B-DFlash
|
||
num_speculative_tokens: 15
|
||
```
|
||
|
||
The shape of `speculative_config` follows vLLM's
|
||
[`SpeculativeConfig`](https://docs.vllm.ai/en/latest/api/vllm/config/speculative.html)
|
||
- `method` picks the algorithm, the remaining keys are method-specific.
|
||
Drafters from [z-lab](https://huggingface.co/z-lab) are paired with
|
||
specific target models; pick the one that matches your target. The
|
||
drafter loads in its native precision regardless of the target's
|
||
`quantization:` setting.
|
||
|
||
Another example - picking a non-default attention backend (e.g. on
|
||
hardware where the default cutlass kernels aren't supported):
|
||
|
||
```yaml
|
||
engine_args:
|
||
attention_backend: TRITON_ATTN
|
||
```
|
||
|
||
#### Multi-node data parallelism
|
||
|
||
`engine_args.data_parallel_size > 1` combined with the
|
||
`local-ai p2p-worker vllm` follower lets a single model span multiple
|
||
GPU nodes. See [vLLM Multi-Node (Data-Parallel)]({{% relref
|
||
"features/distributed-mode#vllm-multi-node-data-parallel" %}})
|
||
for the head/follower configuration and a worked Kimi-K2.6 example.
|
||
|
||
### SGLang
|
||
|
||
[SGLang](https://github.com/sgl-project/sglang) is a fast serving
|
||
framework for LLMs and VLMs with a focus on prefix caching, speculative
|
||
decoding, and multi-modal generation. LocalAI ships a gRPC backend that
|
||
wraps SGLang's async `Engine`, including its native function-call and
|
||
reasoning parsers.
|
||
|
||
#### Setup
|
||
|
||
```yaml
|
||
name: sglang
|
||
backend: sglang
|
||
parameters:
|
||
model: "Qwen/Qwen3-4B"
|
||
template:
|
||
use_tokenizer_template: true
|
||
```
|
||
|
||
The backend will pull the model from HuggingFace on first load.
|
||
|
||
#### Passing arbitrary SGLang options with `engine_args`
|
||
|
||
The same `engine_args:` map that the vLLM backend accepts is also
|
||
honoured by the SGLang backend. Keys are validated against
|
||
[`ServerArgs`](https://github.com/sgl-project/sglang/blob/main/python/sglang/srt/server_args.py)
|
||
- SGLang's central configuration dataclass - and forwarded verbatim to
|
||
`Engine(**kwargs)`. Unknown keys fail at load time with the closest
|
||
valid name as a hint. Unlike vLLM, `ServerArgs` is flat: speculative
|
||
decoding fields are top-level (`speculative_algorithm`,
|
||
`speculative_draft_model_path`, etc.) rather than nested under a
|
||
`speculative_config:` dict.
|
||
|
||
The typed YAML fields shared with vLLM are mapped to their SGLang
|
||
equivalents (`gpu_memory_utilization` → `mem_fraction_static`,
|
||
`enforce_eager` → `disable_cuda_graph`, `tensor_parallel_size` →
|
||
`tp_size`, `max_model_len` → `context_length`). Anything else,
|
||
including all speculative-decoding flags, goes under `engine_args:`.
|
||
|
||
##### Speculative decoding: Gemma 4 with Multi-Token Prediction
|
||
|
||
Google publishes paired "assistant" drafters for every Gemma 4 size.
|
||
The drafters use Multi-Token Prediction (MTP) to propose several
|
||
candidate tokens per target step, which SGLang then verifies in
|
||
parallel. Flags below are transcribed verbatim from the
|
||
[SGLang Gemma 4 cookbook](https://docs.sglang.io/cookbook/autoregressive/Google/Gemma4#speculative-decoding-mtp-server-commands).
|
||
|
||
For consumer GPUs in the 16-24 GB range, use **E4B** (8 B total /
|
||
4 B effective parameters):
|
||
|
||
```yaml
|
||
name: gemma-4-e4b-mtp
|
||
backend: sglang
|
||
parameters:
|
||
model: google/gemma-4-E4B-it
|
||
context_size: 4096
|
||
template:
|
||
use_tokenizer_template: true
|
||
options:
|
||
- tool_parser:gemma4
|
||
- reasoning_parser:gemma4
|
||
engine_args:
|
||
mem_fraction_static: 0.85
|
||
speculative_algorithm: NEXTN
|
||
speculative_draft_model_path: google/gemma-4-E4B-it-assistant
|
||
speculative_num_steps: 5
|
||
speculative_num_draft_tokens: 6
|
||
speculative_eagle_topk: 1
|
||
```
|
||
|
||
For smaller cards (8-12 GB), drop to **E2B** (5 B total / 2 B effective)
|
||
by swapping the model paths to `google/gemma-4-E2B-it` and
|
||
`google/gemma-4-E2B-it-assistant`; the rest of the flags stay the same.
|
||
|
||
`NEXTN` is normalised to `EAGLE` inside `ServerArgs.__post_init__`, so
|
||
either value works - the cookbook uses `NEXTN`. `mem_fraction_static`
|
||
is the share of GPU memory SGLang reserves for the model + KV pool;
|
||
0.85 is the cookbook's default and adapts to whatever single GPU the
|
||
backend is running on.
|
||
|
||
The 31 B dense and 26 B-A4B MoE Gemma 4 variants exist in the same
|
||
cookbook but require `--tp-size 2`, so they're not in the gallery as
|
||
single-GPU recipes.
|
||
|
||
> **SGLang version requirement.** Gemma 4 support landed in SGLang via
|
||
> [PR #21952](https://github.com/sgl-project/sglang/pull/21952). The
|
||
> LocalAI sglang backend pins a release that includes it; if you've
|
||
> overridden the pin to an older version, this recipe will fail with a
|
||
> "model architecture not recognised" error at load time.
|
||
|
||
##### Other speculative algorithms
|
||
|
||
`speculative_algorithm:` also accepts `EAGLE`/`EAGLE3` (paired with an
|
||
EAGLE-style draft head), `DFLASH` (block-diffusion drafters from
|
||
[z-lab](https://huggingface.co/z-lab) for the Qwen3 family), `STANDALONE`
|
||
(a smaller draft LLM verifying a larger target), and `NGRAM` (no draft
|
||
model - pure prefix-history speculation). See SGLang's
|
||
[speculative-decoding docs](https://docs.sglang.io/advanced_features/speculative_decoding.html)
|
||
for the full algorithm matrix.
|
||
|
||
#### Tool calling and reasoning parsers
|
||
|
||
SGLang's native parsers stream `tool_calls` and `reasoning_content`
|
||
inside `ChatDelta` - the LocalAI Python backend wires them up
|
||
per-request rather than via `engine_args:`. Pick a parser by name:
|
||
|
||
```yaml
|
||
options:
|
||
- tool_parser:hermes
|
||
- reasoning_parser:deepseek_r1
|
||
```
|
||
|
||
The full list of registered parsers lives in `sglang.srt.function_call`
|
||
and `sglang.srt.parser.reasoning_parser`.
|
||
|
||
### vllm.cpp
|
||
|
||
[vllm.cpp](https://github.com/mudler/vllm.cpp) is the LocalAI team's C++ port of
|
||
vLLM: the same continuous-batching scheduler, paged KV cache and prefix caching,
|
||
with no Python at inference time. It consumes either a HuggingFace safetensors
|
||
model directory or a `.gguf` file, and applies the model's chat template,
|
||
tool-call parsing and reasoning split engine-side.
|
||
|
||
#### Setup
|
||
|
||
```yaml
|
||
name: vllm-cpp
|
||
backend: vllm-cpp
|
||
parameters:
|
||
model: "Qwen/Qwen3-4B"
|
||
context_size: 8192
|
||
template:
|
||
use_tokenizer_template: true
|
||
```
|
||
|
||
#### Configuring the engine with `engine_args`
|
||
|
||
The same `engine_args:` map the vLLM and SGLang backends accept is honoured
|
||
here, with keys spelled exactly as vLLM's own CLI flags - so a `speculative_config`
|
||
or `kv_transfer_config` block written for vLLM works verbatim. Unknown keys are
|
||
ignored rather than fatal; the engine validates the documents it is handed and
|
||
reports a precise error at load.
|
||
|
||
```yaml
|
||
name: qwen35-a3b
|
||
backend: vllm-cpp
|
||
parameters:
|
||
model: "Qwen/Qwen3.5-A3B"
|
||
context_size: 16384
|
||
template:
|
||
use_tokenizer_template: true
|
||
engine_args:
|
||
# KV cache sizing: num_blocks * block_size tokens of cache.
|
||
block_size: 32
|
||
num_blocks: 1024
|
||
# Concurrency and the per-step chunked-prefill token budget.
|
||
max_num_seqs: 32
|
||
max_num_batched_tokens: 8192
|
||
# Automatic prefix caching. Omit to keep the model's own default
|
||
# (on for dense models, off for hybrid / attention-free ones).
|
||
enable_prefix_caching: true
|
||
# Scheduler admission order: fcfs (default), priority, or lpm
|
||
# (cache-aware longest-prefix-match; needs prefix caching to have any effect).
|
||
scheduling_policy: lpm
|
||
```
|
||
|
||
| Key | Meaning | Default |
|
||
|-----|---------|---------|
|
||
| `block_size` | KV-cache block size, in tokens per block | 32 |
|
||
| `num_blocks` | KV-cache blocks to allocate | 256 |
|
||
| `max_model_len` | Max sequence length; also settable as `context_size` / `max_model_len` | model config |
|
||
| `max_num_seqs` | Max concurrent sequences the scheduler admits | 8 |
|
||
| `max_num_batched_tokens` | Per-step chunked-prefill token budget | per-arch (2048 dense, 4096/8192 MoE) |
|
||
| `enable_prefix_caching` | Automatic prefix caching; `enable_radix_attention` is an accepted alias | model default |
|
||
| `enable_jump_forward` | Jump-forward decoding, which emits grammar-forced tokens without a model step. Only affects constrained requests (`grammar`, JSON schema) | off |
|
||
| `scheduling_policy` | `fcfs`, `priority`, or `lpm` | `fcfs` |
|
||
| `tool_parser` / `reasoning_parser` | Force a parser instead of chat-template auto-detection | auto |
|
||
| `tokenizer_config` | Override the `tokenizer_config.json` the chat template is read from | `<model_dir>/tokenizer_config.json` |
|
||
| `speculative_config` | Speculative decoding (see below) | disabled |
|
||
| `kv_transfer_config` | External KV connector / LMCache (see below) | none |
|
||
|
||
Raising `max_num_batched_tokens` lets more prefill land in a single step, at the
|
||
cost of decode latency for requests queued behind it. The default deliberately
|
||
does not scale with `max_num_seqs`, which is what keeps a large concurrent
|
||
prefill from blowing up the per-step activation on the hybrid architectures.
|
||
|
||
`enable_prefix_caching` and `enable_jump_forward` are tri-state at the engine
|
||
boundary: omitting the key defers to a default (the model's own capability for
|
||
prefix caching, an environment variable for jump forward), while an explicit
|
||
`false` forces the feature off. Those are genuinely different - prefix caching
|
||
defaults *on* for dense models - so write the key only when you mean to override.
|
||
|
||
#### Speculative decoding
|
||
|
||
`speculative_config:` takes the same JSON object as vLLM's
|
||
`--speculative-config`. Three methods are supported.
|
||
|
||
> **Architecture limit.** At the current engine pin, `mtp` and `dflash` are
|
||
> **Qwen3.5 / Qwen3.6 only**. The engine builds a widened speculative KV cache
|
||
> directly for those families rather than through the model registry, so a
|
||
> speculative config on any other architecture (Llama, GLM, Gemma, Mistral, ...)
|
||
> will not work regardless of checkpoint format. `ngram` needs no draft weights
|
||
> and is not subject to this limit.
|
||
|
||
> **Format support.** `mtp` and `dflash` now work from a `.gguf` target as well
|
||
> as safetensors. An MTP head is read from the GGUF's `nextn.*` tensors when the
|
||
> file declares `<arch>.nextn_predict_layers`; a GGUF exported WITHOUT the head
|
||
> (converted with `--no-mtp`, or predating llama.cpp's Qwen3.5 MTP support) is
|
||
> refused at load naming that as the reason. A DFlash draft may itself be a
|
||
> `dflash`-arch GGUF, and the target may be a GGUF too. `ngram` needs no draft
|
||
> weights and works on any format.
|
||
|
||
**MTP** (Multi-Token Prediction) uses a draft head shipped inside the target
|
||
checkpoint's own `mtp.*` tensors, so there is no second model to download. It
|
||
requires a **safetensors** checkpoint - the `mtp.*` tensors do not survive GGUF
|
||
conversion, and an MTP config over a `.gguf` model is rejected at load.
|
||
|
||
```yaml
|
||
engine_args:
|
||
speculative_config:
|
||
method: mtp
|
||
# Optional; defaults to the checkpoint's own head depth, which is
|
||
# usually the right value. Must be a multiple of that depth.
|
||
num_speculative_tokens: 1
|
||
```
|
||
|
||
**DFlash** uses a separate block-diffusion drafter that proposes a whole block
|
||
of tokens in one non-autoregressive forward pass. Unlike MTP, the draft is its
|
||
own checkpoint, so `model:` is **required**:
|
||
|
||
```yaml
|
||
engine_args:
|
||
speculative_config:
|
||
method: dflash
|
||
model: z-lab/Qwen3.6-27B-DFlash
|
||
num_speculative_tokens: 4
|
||
```
|
||
|
||
The draft shares the *target's* `embed_tokens` and `lm_head`, so both must come
|
||
from the same model family and the target must be safetensors.
|
||
|
||
**The engine does not download the draft.** `model:` is resolved, in order,
|
||
as a path as given, then as the last path segment under LocalAI's models
|
||
directory (`z-lab/Qwen3.6-27B-DFlash` → `<models>/Qwen3.6-27B-DFlash`, which is
|
||
what LocalAI's own downloader produces), then as the whole reference under the
|
||
models directory. Install the draft into LocalAI first, or give an absolute path
|
||
to a directory containing `config.json`. If none of those resolve, the load
|
||
fails immediately naming every location that was tried, rather than reporting a
|
||
missing checkpoint from inside the engine.
|
||
|
||
**N-gram** needs no draft model at all - it proposes from the prompt's own
|
||
suffix history. `num_speculative_tokens` is required:
|
||
|
||
```yaml
|
||
engine_args:
|
||
speculative_config:
|
||
method: ngram
|
||
num_speculative_tokens: 4
|
||
prompt_lookup_min: 5
|
||
prompt_lookup_max: 5
|
||
```
|
||
|
||
> **Auto-configuration on import.** When you import a safetensors repository
|
||
> with `backend: vllm-cpp`, LocalAI reads the checkpoint's `config.json` and, if
|
||
> it declares an MTP head (`mtp_num_hidden_layers`), writes
|
||
> `speculative_config: {method: mtp}` into the generated `engine_args` for you.
|
||
> An explicit `speculative_config` in your own config is never overwritten.
|
||
> Importing a DFlash *draft* repository is refused with a warning: a drafter
|
||
> cannot serve on its own, so import the target model and point
|
||
> `speculative_config.model` at the draft.
|
||
|
||
#### External KV cache with LMCache
|
||
|
||
`kv_transfer_config:` takes vLLM's `--kv-transfer-config` JSON and selects an
|
||
external KV-cache connector. The `lm://` LMCache client lets prefill KV be
|
||
stored to and reloaded from a shared `lmcache.v1.server`, so a prefix computed
|
||
by one replica does not have to be recomputed by the next:
|
||
|
||
```yaml
|
||
engine_args:
|
||
kv_transfer_config:
|
||
kv_connector: LMCacheConnector
|
||
kv_role: kv_both # required whenever kv_connector is set
|
||
kv_connector_extra_config:
|
||
host: 127.0.0.1
|
||
port: 65432
|
||
```
|
||
|
||
`kv_role` is one of `kv_producer` (store only), `kv_consumer` (load only), or
|
||
`kv_both`. An unregistered connector name, a missing role, or a malformed
|
||
document fails the load with an explicit error rather than silently running
|
||
without the cache.
|
||
|
||
#### Legacy `options:` list
|
||
|
||
Earlier versions configured this backend through the flat `options:` list, and
|
||
those configs keep working. Every key in the table above is still read from
|
||
there in `key:value` form, and `engine_args` wins on any key set in both:
|
||
|
||
```yaml
|
||
options:
|
||
- max_num_seqs:32
|
||
- enable_prefix_caching:true
|
||
```
|
||
|
||
New configs should prefer `engine_args:`, which is the only place the nested
|
||
`speculative_config` / `kv_transfer_config` documents can be written naturally
|
||
rather than as a single-line JSON string.
|
||
|
||
### Transformers
|
||
|
||
[Transformers](https://huggingface.co/docs/transformers/index) is a State-of-the-art Machine Learning library for PyTorch, TensorFlow, and JAX.
|
||
|
||
LocalAI has a built-in integration with Transformers, and it can be used to run models.
|
||
|
||
This is an extra backend - in the container images (the `extra` images already contains python dependencies for Transformers) is already available and there is nothing to do for the setup.
|
||
|
||
#### Setup
|
||
|
||
Create a YAML file for the model you want to use with `transformers`.
|
||
|
||
To setup a model, you need to just specify the model name in the YAML config file:
|
||
```yaml
|
||
name: transformers
|
||
backend: transformers
|
||
parameters:
|
||
model: "facebook/opt-125m"
|
||
type: AutoModelForCausalLM
|
||
quantization: bnb_4bit # One of: bnb_8bit, bnb_4bit, xpu_4bit, xpu_8bit (optional)
|
||
```
|
||
|
||
The backend will automatically download the required files in order to run the model.
|
||
|
||
#### Parameters
|
||
|
||
##### Type
|
||
|
||
| Type | Description |
|
||
| --- | --- |
|
||
| `AutoModelForCausalLM` | `AutoModelForCausalLM` is a model that can be used to generate sequences. Use it for NVIDIA CUDA and Intel GPU with Intel Extensions for Pytorch acceleration |
|
||
| `OVModelForCausalLM` | for Intel CPU/GPU/NPU OpenVINO Text Generation models |
|
||
| `OVModelForFeatureExtraction` | for Intel CPU/GPU/NPU OpenVINO Embedding acceleration |
|
||
| N/A | Defaults to `AutoModel` |
|
||
|
||
- `OVModelForCausalLM` requires OpenVINO IR [Text Generation](https://huggingface.co/models?library=openvino&pipeline_tag=text-generation) models from Hugging face
|
||
- `OVModelForFeatureExtraction` works with any Safetensors Transformer [Feature Extraction](https://huggingface.co/models?pipeline_tag=feature-extraction&library=transformers,safetensors) model from Huggingface (Embedding Model)
|
||
|
||
Please note that streaming is currently not implemented in `AutoModelForCausalLM` for Intel GPU.
|
||
AMD GPU support is not implemented.
|
||
Although AMD CPU is not officially supported by OpenVINO there are reports that it works: YMMV.
|
||
|
||
##### Embeddings
|
||
Use `embeddings: true` if the model is an embedding model
|
||
|
||
##### Inference device selection
|
||
Transformer backend tries to automatically select the best device for inference, anyway you can override the decision manually overriding with the `main_gpu` parameter.
|
||
|
||
| Inference Engine | Applicable Values |
|
||
| --- | --- |
|
||
| CUDA | `cuda`, `cuda.X` where X is the GPU device like in `nvidia-smi -L` output |
|
||
| OpenVINO | Any applicable value from [Inference Modes](https://docs.openvino.ai/2024/openvino-workflow/running-inference/inference-devices-and-modes.html) like `AUTO`,`CPU`,`GPU`,`NPU`,`MULTI`,`HETERO` |
|
||
|
||
Example for CUDA:
|
||
`main_gpu: cuda.0`
|
||
|
||
Example for OpenVINO:
|
||
`main_gpu: AUTO:-CPU`
|
||
|
||
This parameter applies to both Text Generation and Feature Extraction (i.e. Embeddings) models.
|
||
|
||
##### Inference Precision
|
||
Transformer backend automatically select the fastest applicable inference precision according to the device support.
|
||
CUDA backend can manually enable *bfloat16* if your hardware support it with the following parameter:
|
||
|
||
`f16: true`
|
||
|
||
##### Quantization
|
||
|
||
| Quantization | Description |
|
||
| --- | --- |
|
||
| `bnb_8bit` | 8-bit quantization |
|
||
| `bnb_4bit` | 4-bit quantization |
|
||
| `xpu_8bit` | 8-bit quantization for Intel XPUs |
|
||
| `xpu_4bit` | 4-bit quantization for Intel XPUs |
|
||
|
||
##### Trust Remote Code
|
||
Some models like Microsoft Phi-3 requires external code than what is provided by the transformer library.
|
||
By default it is disabled for security.
|
||
It can be manually enabled with:
|
||
`trust_remote_code: true`
|
||
|
||
##### Maximum Context Size
|
||
Maximum context size in bytes can be specified with the parameter: `context_size`. Do not use values higher than what your model support.
|
||
|
||
Usage example:
|
||
`context_size: 8192`
|
||
|
||
##### Auto Prompt Template
|
||
Usually chat template is defined by the model author in the `tokenizer_config.json` file.
|
||
To enable it use the `use_tokenizer_template: true` parameter in the `template` section.
|
||
|
||
Usage example:
|
||
```
|
||
template:
|
||
use_tokenizer_template: true
|
||
```
|
||
|
||
##### Custom Stop Words
|
||
Stopwords are usually defined in `tokenizer_config.json` file.
|
||
They can be overridden with the `stopwords` parameter in case of need like in llama3-Instruct model.
|
||
|
||
Usage example:
|
||
```
|
||
stopwords:
|
||
- "<|eot_id|>"
|
||
- "<|end_of_text|>"
|
||
```
|
||
|
||
#### Usage
|
||
|
||
Use the `completions` endpoint by specifying the `transformers` model:
|
||
```
|
||
curl http://localhost:8080/v1/completions -H "Content-Type: application/json" -d '{
|
||
"model": "transformers",
|
||
"prompt": "Hello, my name is",
|
||
"temperature": 0.1, "top_p": 0.1
|
||
}'
|
||
```
|
||
|
||
#### Examples
|
||
|
||
##### OpenVINO
|
||
|
||
A model configuration file for openvion and starling model:
|
||
|
||
```yaml
|
||
name: starling-openvino
|
||
backend: transformers
|
||
parameters:
|
||
model: fakezeta/Starling-LM-7B-beta-openvino-int8
|
||
context_size: 8192
|
||
threads: 6
|
||
f16: true
|
||
type: OVModelForCausalLM
|
||
stopwords:
|
||
- <|end_of_turn|>
|
||
- <|endoftext|>
|
||
prompt_cache_path: "cache"
|
||
prompt_cache_all: true
|
||
template:
|
||
chat_message: |
|
||
{{if eq .RoleName "system"}}{{.Content}}<|end_of_turn|>{{end}}{{if eq .RoleName "assistant"}}<|end_of_turn|>GPT4 Correct Assistant: {{.Content}}<|end_of_turn|>{{end}}{{if eq .RoleName "user"}}GPT4 Correct User: {{.Content}}{{end}}
|
||
|
||
chat: |
|
||
{{.Input}}<|end_of_turn|>GPT4 Correct Assistant:
|
||
|
||
completion: |
|
||
{{.Input}}
|
||
```
|