mirror of
https://github.com/ggerganov/llama.cpp.git
synced 2024-11-15 23:39:52 +00:00
38373cfbab
* add --spm-infill option * support --spm-infill * support --spm-infill
48 lines
2.6 KiB
Markdown
48 lines
2.6 KiB
Markdown
# llama.cpp/example/infill
|
|
|
|
This example shows how to use the infill mode with Code Llama models supporting infill mode.
|
|
Currently the 7B and 13B models support infill mode.
|
|
|
|
Infill supports most of the options available in the main example.
|
|
|
|
For further information have a look at the main README.md in llama.cpp/example/main/README.md
|
|
|
|
## Common Options
|
|
|
|
In this section, we cover the most commonly used options for running the `infill` program with the LLaMA models:
|
|
|
|
- `-m FNAME, --model FNAME`: Specify the path to the LLaMA model file (e.g., `models/7B/ggml-model.bin`).
|
|
- `-i, --interactive`: Run the program in interactive mode, allowing you to provide input directly and receive real-time responses.
|
|
- `-n N, --n-predict N`: Set the number of tokens to predict when generating text. Adjusting this value can influence the length of the generated text.
|
|
- `-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.
|
|
- `--spm-infill`: Use Suffix/Prefix/Middle pattern for infill (instead of Prefix/Suffix/Middle) as some models prefer this.
|
|
|
|
## Input Prompts
|
|
|
|
The `infill` program provides several ways to interact with the LLaMA models using input prompts:
|
|
|
|
- `--in-prefix PROMPT_BEFORE_CURSOR`: Provide the prefix directly as a command-line option.
|
|
- `--in-suffix PROMPT_AFTER_CURSOR`: Provide the suffix directly as a command-line option.
|
|
- `--interactive-first`: Run the program in interactive mode and wait for input right away. (More on this below.)
|
|
|
|
## Interaction
|
|
|
|
The `infill` program offers a seamless way to interact with LLaMA models, allowing users to receive real-time infill suggestions. The interactive mode can be triggered using `--interactive`, and `--interactive-first`
|
|
|
|
### Interaction Options
|
|
|
|
- `-i, --interactive`: Run the program in interactive mode, allowing users to get real time code suggestions from model.
|
|
- `--interactive-first`: Run the program in interactive mode and immediately wait for user input before starting the text generation.
|
|
- `--color`: Enable colorized output to differentiate visually distinguishing between prompts, user input, and generated text.
|
|
|
|
### Example
|
|
|
|
Download a model that supports infill, for example CodeLlama:
|
|
```console
|
|
scripts/hf.sh --repo TheBloke/CodeLlama-13B-GGUF --file codellama-13b.Q5_K_S.gguf --outdir models
|
|
```
|
|
|
|
```bash
|
|
./llama-infill -t 10 -ngl 0 -m models/codellama-13b.Q5_K_S.gguf -c 4096 --temp 0.7 --repeat_penalty 1.1 -n 20 --in-prefix "def helloworld():\n print(\"hell" --in-suffix "\n print(\"goodbye world\")\n "
|
|
```
|