aboutsummaryrefslogtreecommitdiff
path: root/examples/chatLLaMa
diff options
context:
space:
mode:
Diffstat (limited to 'examples/chatLLaMa')
-rwxr-xr-xexamples/chatLLaMa2
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/chatLLaMa b/examples/chatLLaMa
index 97ababb..4265d7b 100755
--- a/examples/chatLLaMa
+++ b/examples/chatLLaMa
@@ -13,7 +13,7 @@ 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 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}"
+GEN_OPTIONS="${GEN_OPTIONS:---ctx_size 2048 --temp 0.7 --top_k 40 --top_p 0.5 --repeat_last_n 256 --batch_size 1024 --repeat_penalty 1.17647}"
# shellcheck disable=SC2086 # Intended splitting of GEN_OPTIONS
./main $GEN_OPTIONS \