Skip to content

Commit e587f7c

Browse files
committed
Include n_predict to 2048 in examples/chatLLaMa
1 parent abd92dd commit e587f7c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

examples/chatLLaMa

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,17 +2,17 @@
22

33
cd "$(dirname "$0")/.." || exit
44

5-
MODEL="./models/13B/ggml-model-q4_0.bin"
5+
MODEL="${MODEL:-./models/13B/ggml-model-q4_0.bin}"
66
USER_NAME="${USER_NAME:-User}"
77
AI_NAME="${AI_NAME:-ChatLLaMa}"
88

99
# Adjust to the number of CPU cores you want to use.
1010
N_THREAD="${N_THREAD:-8}"
1111
# Number of tokens to predict (made it larger than default because we want a long interaction)
12-
N_PREDICTS="${N_PREDICTS:-1024}"
12+
N_PREDICTS="${N_PREDICTS:-2048}"
1313

1414
# Note: you can also override the generation options by specifying them on the command line:
15-
# For example, override the context size by doing: ./chatLLaMa --ctx_size 2048
15+
# For example, override the context size by doing: ./chatLLaMa --ctx_size 1024
1616
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}"
1717

1818
# shellcheck disable=SC2086 # Intended splitting of GEN_OPTIONS

0 commit comments

Comments
 (0)