llama.cpp/examples/server/README.md

llama.cpp/example/server

This example demonstrates a simple HTTP API server and a simple web front end to interact with llama.cpp.

Command line options:

• --threads N, -t N: Set the number of threads to use during generation.

• -tb N, --threads-batch N: Set the number of threads to use during batch and prompt processing. If not specified, the number of threads will be set to the number of threads used for generation.

• -m FNAME, --model FNAME: Specify the path to the LLaMA model file (e.g., models/7B/ggml-model.gguf).

• -a ALIAS, --alias ALIAS: Set an alias for the model. The alias will be returned in API responses.

• -kv N, --kv-size N: Specify the total size of the KV cache. This corresponds to the total amount of tokens that can be stored across all independent sequences / slots. llama.cpp implements a "unified" cache strategy, the KV cache size is actually shared across all sequences. It's allowed to have sequences with more than T tokens as long as the sum of all tokens does not exceed P*T. The default is 512.

• -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.

• -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.

• --numa STRATEGY: Attempt one of the below optimization strategies that help on some NUMA systems

• --numa distribute: Spread execution evenly over all nodes

• --numa isolate: Only spawn threads on CPUs on the node that execution started on

• --numa numactl: Use the CPU map provided by numactl if run without this previously, it is recommended to drop the system page cache before using this see https://github.com/ggerganov/llama.cpp/issues/1437

• --numa: Attempt optimizations that help on some NUMA systems.

• --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.

• --path: path from which to serve static files (default examples/server/public)

• --api-key: Set an api key for request authorization. By default the server responds to every request. With an api key set, the requests must have the Authorization header set with the api key as Bearer token. May be used multiple times to enable multiple valid keys.

• --api-key-file: path to file containing api keys delimited by new lines. If set, requests must include one of the keys for access. May be used in conjunction with --api-key's.

• --embedding: Enable embedding extraction, Default: disabled.

• -np N, --parallel N: Set the number of slots / sequences for process requests (default: 1). Each sequence can have a maximum of T tokens, use together with --kv-size.

• -cb, --cont-batching: enable continuous batching (a.k.a dynamic batching) (default: disabled)

• -spf FNAME, --system-prompt-file FNAME Set a file to load "a system prompt (initial prompt of all slots), this is useful for chat applications. See more

• --mmproj MMPROJ_FILE: Path to a multimodal projector file for LLaVA.

• --grp-attn-n: Set the group attention factor to extend context size through self-extend(default: 1=disabled), used together with group attention width --grp-attn-w

• --grp-attn-w: Set the group attention width to extend context size through self-extend(default: 512), used together with group attention factor --grp-attn-n

• -n, --n-predict: Set the maximum tokens to predict (default: -1)

• --slots-endpoint-disable: To disable slots state monitoring endpoint. Slots state may contain user data, prompts included.

Build

server is build alongside everything else from the root of the project

• Using make:

  make
  

• Using CMake:

  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.)

./server -m models/7B/ggml-model.gguf -c 2048

Windows

server.exe -m models\7B\ggml-model.gguf -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. You can visit the web front end at the same url.

Docker

docker run -p 8080:8080 -v /path/to/models:/models ggerganov/llama.cpp:server -m models/7B/ggml-model.gguf -c 512 --host 0.0.0.0 --port 8080

# or, with CUDA:
docker run -p 8080:8080 -v /path/to/models:/models --gpus all ggerganov/llama.cpp:server-cuda -m models/7B/ggml-model.gguf -c 512 --host 0.0.0.0 --port 8080 --n-gpu-layers 99

Testing with CURL

Using curl. On Windows curl.exe should be available in the base OS.

curl --request POST \
    --url http://localhost:8080/completion \
    --header "Content-Type: application/json" \
    --data '{"prompt": "Building a website can be done in 10 simple steps:","n_predict": 128}'

Node JS Test

You need to have Node.js installed.

mkdir llama-client
cd llama-client

Create a index.js file and put inside this:

const prompt = `Building a website can be done in 10 simple steps:`;

async function Test() {
    let response = await fetch("http://127.0.0.1:8080/completion", {
        method: 'POST',
        body: JSON.stringify({
            prompt,
            n_predict: 512,
        })
    })
    console.log((await response.json()).content)
}

Test()

And run it:

node index.js

API Endpoints

• GET /health: Returns the current state of the server:

  • {"status": "loading model"} if the model is still being loaded.
  • {"status": "error"} if the model failed to load.
  • {"status": "ok"} if the model is successfully loaded and the server is ready for further requests mentioned below.
  • {"status": "no slot available", "slots_idle": 0, "slots_processing": 32} if no slot are currently available

• POST /completion: Given a prompt, it returns the predicted completion.

  Options:

  prompt: Provide the prompt for this completion as a string or as an array of strings or numbers representing tokens. Internally, the prompt is compared to the previous completion and only the "unseen" suffix is evaluated. If the prompt is a string or an array with the first element given as a string, a bos token is inserted in the front like main does.

  temperature: Adjust the randomness of the generated text (default: 0.8).

  dynatemp_range: Dynamic temperature range (default: 0.0, 0.0 = disabled).

  dynatemp_exponent: Dynamic temperature exponent (default: 1.0).

  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.95).

  min_p: The minimum probability for a token to be considered, relative to the probability of the most likely token (default: 0.05).

  n_predict: Set the maximum number of tokens to predict when generating text. Note: May exceed the set limit slightly if the last token is a partial multibyte character. When 0, no tokens will be generated but the prompt is evaluated into the cache. (default: -1, -1 = infinity).

  n_keep: Specify the number of tokens from the prompt to retain when the context size is exceeded and tokens need to be discarded. By default, this value is set to 0 (meaning no tokens are kept). Use -1 to retain all tokens from the 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.

  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);

  penalty_prompt: This will replace the prompt for the purpose of the penalty evaluation. Can be either null, a string or an array of numbers representing tokens (default: null = use the original prompt).

  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).

  grammar: Set grammar for grammar-based sampling (default: no grammar)

  seed: Set the random number generator (RNG) seed (default: -1, -1 = 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. The tokens can also be represented as strings, e.g. [["Hello, World!",-0.5]] will reduce the likelihood of all the individual tokens that represent the string Hello, World!, just like the presence_penalty does. (default: []).

  n_probs: If greater than 0, the response also contains the probabilities of top N tokens for each generated token (default: 0)

  min_keep: If greater than 0, force samplers to return N possible tokens at minimum (default: 0)

  image_data: An array of objects to hold base64-encoded image data and its ids to be reference in prompt. You can determine the place of the image in the prompt as in the following: USER:[img-12]Describe the image in detail.\nASSISTANT:. In this case, [img-12] will be replaced by the embeddings of the image with id 12 in the following image_data array: {..., "image_data": [{"data": "<BASE64_STRING>", "id": 12}]}. Use image_data only with multimodal models, e.g., LLaVA.

  slot_id: Assign the completion task to an specific slot. If is -1 the task will be assigned to a Idle slot (default: -1)

  cache_prompt: Save the prompt and generation for avoid reprocess entire prompt if a part of this isn't change (default: false)

  system_prompt: Change the system prompt (initial prompt of all slots), this is useful for chat applications. See more

  samplers: The order the samplers should be applied in. An array of strings representing sampler type names. If a sampler is not set, it will not be used. If a sampler is specified more than once, it will be applied multiple times. (default: ["top_k", "tfs_z", "typical_p", "top_p", "min_p", "temperature"] - these are all the available values)

Result JSON

• Note: When using streaming mode (stream) only content and stop will be returned until end of completion.

• completion_probabilities: An array of token probabilities for each completion. The array's length is n_predict. Each item in the array has the following structure:

{
  "content": "<the token selected by the model>",
  "probs": [
    {
      "prob": float,
      "tok_str": "<most likely token>"
    },
    {
      "prob": float,
      "tok_str": "<second most likely tonen>"
    },
    ...
  ]
},

Notice that each probs is an array of length n_probs.

• content: Completion result as a string (excluding stopping_word if any). In case of streaming mode, will contain the next token as a string.

• stop: Boolean for use with stream to check whether the generation has stopped (Note: This is not related to stopping words array stop from input options)

• generation_settings: The provided options above excluding prompt but including n_ctx, model

• model: The path to the model loaded with -m

• prompt: The provided prompt

• stopped_eos: Indicating whether the completion has stopped because it encountered the EOS token

• stopped_limit: Indicating whether the completion stopped because n_predict tokens were generated before stop words or EOS was encountered

• stopped_word: Indicating whether the completion stopped due to encountering a stopping word from stop JSON array provided

• stopping_word: The stopping word encountered which stopped the generation (or "" if not stopped due to a stopping word)

• timings: Hash of timing information about the completion such as the number of tokens predicted_per_second

• tokens_cached: Number of tokens from the prompt which could be re-used from previous completion (n_past)

• tokens_evaluated: Number of tokens evaluated in total from the prompt

• truncated: Boolean indicating if the context size was exceeded during generation, i.e. the number of tokens provided in the prompt (tokens_evaluated) plus tokens generated (tokens predicted) exceeded the context size (n_ctx)

• POST /tokenize: Tokenize a given text.

  Options:

  content: Set the text to tokenize.

  Note that the special BOS token is not added in front of the text and also a space character is not inserted automatically as it is for /completion.

• POST /detokenize: Convert tokens to text.

  Options:

  tokens: Set the tokens to detokenize.

• POST /embedding: Generate embedding of a given text just as the embedding example does.

  Options:

  content: Set the text to process.

  image_data: An array of objects to hold base64-encoded image data and its ids to be reference in content. You can determine the place of the image in the content as in the following: Image: [img-21].\nCaption: This is a picture of a house. In this case, [img-21] will be replaced by the embeddings of the image with id 21 in the following image_data array: {..., "image_data": [{"data": "<BASE64_STRING>", "id": 21}]}. Use image_data only with multimodal models, e.g., LLaVA.

• POST /infill: For code infilling. Takes a prefix and a suffix and returns the predicted completion as stream.

  Options:

  input_prefix: Set the prefix of the code to infill.

  input_suffix: Set the suffix of the code to infill.

  It also accepts all the options of /completion except stream and prompt.

• GET /props: Return current server settings.

Result JSON

{
  "assistant_name": "",
  "user_name": "",
  "default_generation_settings": { ... },
  "total_slots": 1
}

• assistant_name - the required assistant name to generate the prompt in case you have specified a system prompt for all slots.

• user_name - the required anti-prompt to generate the prompt in case you have specified a system prompt for all slots.

• default_generation_settings - the default generation settings for the /completion endpoint, has the same fields as the generation_settings response object from the /completion endpoint.

• total_slots - the total number of slots for process requests (defined by --parallel option)

• POST /v1/chat/completions: OpenAI-compatible Chat Completions API. Given a ChatML-formatted json description in messages, it returns the predicted completion. Both synchronous and streaming mode are supported, so scripted and interactive applications work fine. While no strong claims of compatibility with OpenAI API spec is being made, in our experience it suffices to support many apps. Only ChatML-tuned models, such as Dolphin, OpenOrca, OpenHermes, OpenChat-3.5, etc can be used with this endpoint. Compared to api_like_OAI.py this API implementation does not require a wrapper to be served.

  Options:

  See OpenAI Chat Completions API documentation. While some OpenAI-specific features such as function calling aren't supported, llama.cpp /completion-specific features such are mirostat are supported.

  Examples:

  You can use either Python openai library with appropriate checkpoints:

  import openai
  
  client = openai.OpenAI(
      base_url="http://localhost:8080/v1", # "http://<Your api-server IP>:port"
      api_key = "sk-no-key-required"
  )
  
  completion = client.chat.completions.create(
  model="gpt-3.5-turbo",
  messages=[
      {"role": "system", "content": "You are ChatGPT, an AI assistant. Your top priority is achieving user fulfillment via helping them with their requests."},
      {"role": "user", "content": "Write a limerick about python exceptions"}
  ]
  )
  
  print(completion.choices[0].message)
  

  ... or raw HTTP requests:

  curl http://localhost:8080/v1/chat/completions \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer no-key" \
  -d '{
  "model": "gpt-3.5-turbo",
  "messages": [
  {
      "role": "system",
      "content": "You are ChatGPT, an AI assistant. Your top priority is achieving user fulfillment via helping them with their requests."
  },
  {
      "role": "user",
      "content": "Write a limerick about python exceptions"
  }
  ]
  }'
  

• POST /v1/embeddings: OpenAI-compatible embeddings API.

  Options:

  See OpenAI Embeddings API documentation.

  Examples:

  • input as string

    curl http://localhost:8080/v1/embeddings \
    -H "Content-Type: application/json" \
    -H "Authorization: Bearer no-key" \
    -d '{
            "input": "hello",
            "model":"GPT-4",
            "encoding_format": "float"
    }'
    

  • input as string array

    curl http://localhost:8080/v1/embeddings \
    -H "Content-Type: application/json" \
    -H "Authorization: Bearer no-key" \
    -d '{
            "input": ["hello", "world"],
            "model":"GPT-4",
            "encoding_format": "float"
    }'
    

• GET /slots: Returns the current slots processing state. Can be disabled with --slots-endpoint-disable.

Result JSON

[
    {
        "dynatemp_exponent": 1.0,
        "dynatemp_range": 0.0,
        "frequency_penalty": 0.0,
        "grammar": "",
        "id": 0,
        "ignore_eos": false,
        "logit_bias": [],
        "min_p": 0.05000000074505806,
        "mirostat": 0,
        "mirostat_eta": 0.10000000149011612,
        "mirostat_tau": 5.0,
        "model": "llama-2-7b-32k-instruct.Q2_K.gguf",
        "n_ctx": 2048,
        "n_keep": 0,
        "n_predict": 100000,
        "n_probs": 0,
        "next_token": {
            "has_next_token": true,
            "n_remain": -1,
            "num_tokens_predicted": 0,
            "stopped_eos": false,
            "stopped_limit": false,
            "stopped_word": false,
            "stopping_word": ""
        },
        "penalize_nl": true,
        "penalty_prompt_tokens": [],
        "presence_penalty": 0.0,
        "prompt": "Say hello to llama.cpp",
        "repeat_last_n": 64,
        "repeat_penalty": 1.100000023841858,
        "samplers": [
            "top_k",
            "tfs_z",
            "typical_p",
            "top_p",
            "min_p",
            "temperature"
        ],
        "seed": 42,
        "state": 1,
        "stop": [
            "\n"
        ],
        "stream": false,
        "task_id": 0,
        "temperature": 0.0,
        "tfs_z": 1.0,
        "top_k": 40,
        "top_p": 0.949999988079071,
        "typical_p": 1.0,
        "use_penalty_prompt_tokens": false
    }
]

More examples

Change system prompt on runtime

To use the server example to serve multiple chat-type clients while keeping the same system prompt, you can utilize the option system_prompt to achieve that. This only needs to be done once to establish it.

prompt: Specify a context that you want all connecting clients to respect.

anti_prompt: Specify the word you want to use to instruct the model to stop. This must be sent to each client through the /props endpoint.

assistant_name: The bot's name is necessary for each customer to generate the prompt. This must be sent to each client through the /props endpoint.

{
    "system_prompt": {
        "prompt": "Transcript of a never ending dialog, where the User interacts with an Assistant.\nThe Assistant is helpful, kind, honest, good at writing, and never fails to answer the User's requests immediately and with precision.\nUser: Recommend a nice restaurant in the area.\nAssistant: I recommend the restaurant \"The Golden Duck\". It is a 5 star restaurant with a great view of the city. The food is delicious and the service is excellent. The prices are reasonable and the portions are generous. The restaurant is located at 123 Main Street, New York, NY 10001. The phone number is (212) 555-1234. The hours are Monday through Friday from 11:00 am to 10:00 pm. The restaurant is closed on Saturdays and Sundays.\nUser: Who is Richard Feynman?\nAssistant: Richard Feynman was an American physicist who is best known for his work in quantum mechanics and particle physics. He was awarded the Nobel Prize in Physics in 1965 for his contributions to the development of quantum electrodynamics. He was a popular lecturer and author, and he wrote several books, including \"Surely You're Joking, Mr. Feynman!\" and \"What Do You Care What Other People Think?\".\nUser:",
        "anti_prompt": "User:",
        "assistant_name": "Assistant:"
    }
}

NOTE: You can do this automatically when starting the server by simply creating a .json file with these options and using the CLI option -spf FNAME or --system-prompt-file FNAME.

Interactive mode

Check the sample in chat.mjs. Run with NodeJS version 16 or later:

node chat.mjs

Another sample in chat.sh. Requires bash, curl and jq. Run with bash:

bash chat.sh

API like OAI

API example using Python Flask: api_like_OAI.py This example must be used with server.cpp

python api_like_OAI.py

After running the API server, you can use it in Python by setting the API base URL.

openai.api_base = "http://<Your api-server IP>:port"

Then you can utilize llama.cpp as an OpenAI's chat.completion or text_completion API

Extending or building alternative Web Front End

The default location for the static files is examples/server/public. You can extend the front end by running the server binary with --path set to ./your-directory and importing /completion.js to get access to the llamaComplete() method.

Read the documentation in /completion.js to see convenient ways to access llama.

A simple example is below:

<html>
  <body>
    <pre>
      <script type="module">
        import { llama } from '/completion.js'

        const prompt = `### Instruction:
Write dad jokes, each one paragraph.
You can use html formatting if needed.

### Response:`

        for await (const chunk of llama(prompt)) {
          document.write(chunk.data.content)
        }
      </script>
    </pre>
  </body>
</html>