diff options
| author | Georgi Gerganov <ggerganov@gmail.com> | 2024-01-06 11:40:24 +0200 |
|---|---|---|
| committer | Georgi Gerganov <ggerganov@gmail.com> | 2024-01-06 11:40:24 +0200 |
| commit | 96e80dabc6e73ff68b09b68947b1fc25883c5094 (patch) | |
| tree | f10072b2e92eb62810308b2202b04e300b7f3910 | |
| parent | eec22a1c6378d9a013943cbddb4330c0da621442 (diff) | |
examples : improve base-translate.sh script (#4783)
| -rwxr-xr-x | examples/base-translate.sh | 15 |
1 files changed, 10 insertions, 5 deletions
diff --git a/examples/base-translate.sh b/examples/base-translate.sh index 50fba025..00dedd0d 100755 --- a/examples/base-translate.sh +++ b/examples/base-translate.sh @@ -8,14 +8,19 @@ # cd llama.cpp # make -j # -# ./examples/base-translate.sh <model-base> "<text>" +# ./examples/base-translate.sh <model-base> "<text>" [extra-main-args] # -if [ $# -ne 2 ]; then - echo "Usage: ./base-translate.sh <model-base> \"<text>\"" +if [ $# -lt 2 ]; then + echo "Usage: ./base-translate.sh <model-base> \"<text>\" [extra-main-args]" exit 1 fi +eargs="" +if [ $# -gt 2 ]; then + eargs="${@:3}" +fi + ftmp="__llama.cpp_example_tmp__.txt" trap "rm -f $ftmp" EXIT @@ -52,5 +57,5 @@ echo "$2 model=$1 -# generate the most likely continuation, run on the CPU until the string "===" is found -./main -m $model -f $ftmp -n 64 --temp 0 --repeat-penalty 1.0 --no-penalize-nl -ngl 0 -r "===" +# generate the most likely continuation until the string "===" is found +./main -m $model -f $ftmp -n 64 --temp 0 --repeat-penalty 1.0 --no-penalize-nl -r "===" $eargs |