Include n_predict to 2048 in examples/chatLLaMa

This commit is contained in:
Jean-Christophe Hoelt 2023-03-19 18:27:54 +02:00
parent b8c383a9b9
commit 1b8f8ad0ba
No known key found for this signature in database
GPG key ID: 838AF264602A1B37

View file

@ -2,17 +2,17 @@
cd "$(dirname "$0")/.." || exit
MODEL="./models/13B/ggml-model-q4_0.bin"
MODEL="${MODEL:-./models/13B/ggml-model-q4_0.bin}"
USER_NAME="${USER_NAME:-User}"
AI_NAME="${AI_NAME:-ChatLLaMa}"
# Adjust to the number of CPU cores you want to use.
N_THREAD="${N_THREAD:-8}"
# Number of tokens to predict (made it larger than default because we want a long interaction)
N_PREDICTS="${N_PREDICTS:-1024}"
N_PREDICTS="${N_PREDICTS:-2048}"
# Note: you can also override the generation options by specifying them on the command line:
# For example, override the context size by doing: ./chatLLaMa --ctx_size 2048
# For example, override the context size by doing: ./chatLLaMa --ctx_size 1024
GEN_OPTIONS="${GEN_OPTIONS:---ctx_size 2048 --temp 0.7 --top_k 40 --top_p 0.5 --repeat_last_n 256 --repeat_penalty 1.17647}"
# shellcheck disable=SC2086 # Intended splitting of GEN_OPTIONS