mirror of
https://github.com/ggerganov/llama.cpp.git
synced 2024-12-25 19:04:35 +00:00
794db3e7b9
A major rewrite for the server example. Note that if you have built something on the previous server API, it will probably be incompatible. Check out the examples for how a typical chat app could work. This took a lot of effort, there are 24 PR's closed in the submitter's repo alone, over 160 commits and a lot of comments and testing. Summary of the changes: - adds missing generation parameters: tfs_z, typical_p, repeat_last_n, repeat_penalty, presence_penalty, frequency_penalty, mirostat, penalize_nl, seed, ignore_eos - applies missing top k sampler - removes interactive mode/terminal-like behavior, removes exclude parameter - moves threads and batch size to server command-line parameters - adds LoRA loading and matches command line parameters with main example - fixes stopping on EOS token and with the specified token amount with n_predict - adds server timeouts, host, and port settings - adds expanded generation complete response; adds generation settings, stop reason, prompt truncated, model used, and final text - sets defaults for unspecified parameters between requests - removes /next-token endpoint and as_loop parameter, adds stream parameter and server-sent events for streaming - adds CORS headers to responses - adds request logging, exception printing and optional verbose logging - adds better stopping words handling when matching multiple tokens and while streaming, or when it finishes on a partial stop string - adds printing an error when it can't bind to the host/port specified - fixes multi-byte character handling and replaces invalid UTF-8 characters on responses - prints timing and build info on startup - adds logit bias to request parameters - removes embedding mode - updates documentation; adds streaming Node.js and Bash examples - fixes code formatting - sets server threads to 1 since the current global state doesn't work well with simultaneous requests - adds truncation of the input prompt and better context reset - removes token limit from the input prompt - significantly simplified the logic and removed a lot of variables --------- Co-authored-by: anon998 <131767832+anon998@users.noreply.github.com> Co-authored-by: Henri Vasserman <henv@hot.ee> Co-authored-by: Felix Hellmann <privat@cirk2.de> Co-authored-by: Johannes Gäßler <johannesg@5d6.de> Co-authored-by: Lesaun Harvey <Lesaun@gmail.com>
184 lines
7.7 KiB
Markdown
184 lines
7.7 KiB
Markdown
# llama.cpp/example/server
|
|
|
|
This example demonstrates a simple HTTP API server to interact with llama.cpp.
|
|
|
|
Command line options:
|
|
|
|
- `--threads N`, `-t N`: Set the number of threads to use during computation.
|
|
- `-m FNAME`, `--model FNAME`: Specify the path to the LLaMA model file (e.g., `models/7B/ggml-model.bin`).
|
|
- `-m ALIAS`, `--alias ALIAS`: Set an alias for the model. The alias will be returned in API responses.
|
|
- `-c N`, `--ctx-size N`: Set the size of the prompt context. The default is 512, but LLaMA models were built with a context of 2048, which will provide better results for longer input/inference.
|
|
- `-ngl N`, `--n-gpu-layers N`: When compiled with appropriate support (currently CLBlast or cuBLAS), this option allows offloading some layers to the GPU for computation. Generally results in increased performance.
|
|
- `-mg i, --main-gpu i`: When using multiple GPUs this option controls which GPU is used for small tensors for which the overhead of splitting the computation across all GPUs is not worthwhile. The GPU in question will use slightly more VRAM to store a scratch buffer for temporary results. By default GPU 0 is used. Requires cuBLAS.
|
|
- `-ts SPLIT, --tensor-split SPLIT`: When using multiple GPUs this option controls how large tensors should be split across all GPUs. `SPLIT` is a comma-separated list of non-negative values that assigns the proportion of data that each GPU should get in order. For example, "3,2" will assign 60% of the data to GPU 0 and 40% to GPU 1. By default the data is split in proportion to VRAM but this may not be optimal for performance. Requires cuBLAS.
|
|
- `-lv, --low-vram`: Do not allocate a VRAM scratch buffer for holding temporary results. Reduces VRAM usage at the cost of performance, particularly prompt processing speed. Requires cuBLAS.
|
|
- `-b N`, `--batch-size N`: Set the batch size for prompt processing. Default: `512`.
|
|
- `--memory-f32`: Use 32-bit floats instead of 16-bit floats for memory key+value. Not recommended.
|
|
- `--mlock`: Lock the model in memory, preventing it from being swapped out when memory-mapped.
|
|
- `--no-mmap`: Do not memory-map the model. By default, models are mapped into memory, which allows the system to load only the necessary parts of the model as needed.
|
|
- `--lora FNAME`: Apply a LoRA (Low-Rank Adaptation) adapter to the model (implies --no-mmap). This allows you to adapt the pretrained model to specific tasks or domains.
|
|
- `--lora-base FNAME`: Optional model to use as a base for the layers modified by the LoRA adapter. This flag is used in conjunction with the `--lora` flag, and specifies the base model for the adaptation.
|
|
- `-to N`, `--timeout N`: Server read/write timeout in seconds. Default `600`.
|
|
- `--host`: Set the hostname or ip address to listen. Default `127.0.0.1`.
|
|
- `--port`: Set the port to listen. Default: `8080`.
|
|
|
|
## Build
|
|
|
|
Build llama.cpp with server from repository root with either make or CMake.
|
|
|
|
- Using `make`:
|
|
|
|
```bash
|
|
LLAMA_BUILD_SERVER=1 make
|
|
```
|
|
|
|
- Using `CMake`:
|
|
|
|
```bash
|
|
mkdir build-server
|
|
cd build-server
|
|
cmake -DLLAMA_BUILD_SERVER=ON ..
|
|
cmake --build . --config Release
|
|
```
|
|
|
|
## Quick Start
|
|
|
|
To get started right away, run the following command, making sure to use the correct path for the model you have:
|
|
|
|
### Unix-based systems (Linux, macOS, etc.):
|
|
|
|
```bash
|
|
./server -m models/7B/ggml-model.bin -c 2048
|
|
```
|
|
|
|
### Windows:
|
|
|
|
```powershell
|
|
server.exe -m models\7B\ggml-model.bin -c 2048
|
|
```
|
|
|
|
The above command will start a server that by default listens on `127.0.0.1:8080`.
|
|
You can consume the endpoints with Postman or NodeJS with axios library.
|
|
|
|
## Testing with CURL
|
|
|
|
Using [curl](https://curl.se/). On Windows `curl.exe` should be available in the base OS.
|
|
|
|
```sh
|
|
curl --request POST \
|
|
--url http://localhost:8080/completion \
|
|
--data '{"prompt": "Building a website can be done in 10 simple steps:","n_predict": 128}'
|
|
```
|
|
|
|
## Node JS Test
|
|
|
|
You need to have [Node.js](https://nodejs.org/en) installed.
|
|
|
|
```bash
|
|
mkdir llama-client
|
|
cd llama-client
|
|
npm init
|
|
npm install axios
|
|
```
|
|
|
|
Create a index.js file and put inside this:
|
|
|
|
```javascript
|
|
const axios = require("axios");
|
|
|
|
const prompt = `Building a website can be done in 10 simple steps:`;
|
|
|
|
async function Test() {
|
|
let result = await axios.post("http://127.0.0.1:8080/completion", {
|
|
prompt,
|
|
n_predict: 512,
|
|
});
|
|
|
|
// the response is received until completion finish
|
|
console.log(result.data.content);
|
|
}
|
|
|
|
Test();
|
|
```
|
|
|
|
And run it:
|
|
|
|
```bash
|
|
node .
|
|
```
|
|
|
|
## API Endpoints
|
|
|
|
- **POST** `/completion`: Given a prompt, it returns the predicted completion.
|
|
|
|
*Options:*
|
|
|
|
`temperature`: Adjust the randomness of the generated text (default: 0.8).
|
|
|
|
`top_k`: Limit the next token selection to the K most probable tokens (default: 40).
|
|
|
|
`top_p`: Limit the next token selection to a subset of tokens with a cumulative probability above a threshold P (default: 0.9).
|
|
|
|
`n_predict`: Set the number of tokens to predict when generating text. **Note:** May exceed the set limit slightly if the last token is a partial multibyte character. (default: 128, -1 = infinity).
|
|
|
|
`n_keep`: Specify the number of tokens from the initial prompt to retain when the model resets its internal context.
|
|
By default, this value is set to 0 (meaning no tokens are kept). Use `-1` to retain all tokens from the initial prompt.
|
|
|
|
`stream`: It allows receiving each predicted token in real-time instead of waiting for the completion to finish. To enable this, set to `true`.
|
|
|
|
`prompt`: Provide a prompt. Internally, the prompt is compared, and it detects if a part has already been evaluated, and the remaining part will be evaluate.
|
|
|
|
`stop`: Specify a JSON array of stopping strings.
|
|
These words will not be included in the completion, so make sure to add them to the prompt for the next iteration (default: []).
|
|
|
|
`tfs_z`: Enable tail free sampling with parameter z (default: 1.0, 1.0 = disabled).
|
|
|
|
`typical_p`: Enable locally typical sampling with parameter p (default: 1.0, 1.0 = disabled).
|
|
|
|
`repeat_penalty`: Control the repetition of token sequences in the generated text (default: 1.1).
|
|
|
|
`repeat_last_n`: Last n tokens to consider for penalizing repetition (default: 64, 0 = disabled, -1 = ctx-size).
|
|
|
|
`penalize_nl`: Penalize newline tokens when applying the repeat penalty (default: true).
|
|
|
|
`presence_penalty`: Repeat alpha presence penalty (default: 0.0, 0.0 = disabled).
|
|
|
|
`frequency_penalty`: Repeat alpha frequency penalty (default: 0.0, 0.0 = disabled);
|
|
|
|
`mirostat`: Enable Mirostat sampling, controlling perplexity during text generation (default: 0, 0 = disabled, 1 = Mirostat, 2 = Mirostat 2.0).
|
|
|
|
`mirostat_tau`: Set the Mirostat target entropy, parameter tau (default: 5.0).
|
|
|
|
`mirostat_eta`: Set the Mirostat learning rate, parameter eta (default: 0.1).
|
|
|
|
`seed`: Set the random number generator (RNG) seed (default: -1, < 0 = random seed).
|
|
|
|
`ignore_eos`: Ignore end of stream token and continue generating (default: false).
|
|
|
|
`logit_bias`: Modify the likelihood of a token appearing in the generated text completion. For example, use `"logit_bias": [[15043,1.0]]` to increase the likelihood of the token 'Hello', or `"logit_bias": [[15043,-1.0]]` to decrease its likelihood. Setting the value to false, `"logit_bias": [[15043,false]]` ensures that the token `Hello` is never produced (default: []).
|
|
|
|
- **POST** `/tokenize`: Tokenize a given text.
|
|
|
|
*Options:*
|
|
|
|
`content`: Set the text to tokenize.
|
|
|
|
## More examples
|
|
|
|
### Interactive mode
|
|
|
|
Check the sample in [chat.mjs](chat.mjs).
|
|
Run with NodeJS version 16 or later:
|
|
|
|
```sh
|
|
node chat.mjs
|
|
```
|
|
|
|
Another sample in [chat.sh](chat.sh).
|
|
Requires [bash](https://www.gnu.org/software/bash/), [curl](https://curl.se) and [jq](https://jqlang.github.io/jq/).
|
|
Run with bash:
|
|
|
|
```sh
|
|
bash chat.sh
|
|
```
|