| # 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 " | |
| ``` | |