Explorar o código

examples : improve base-translate.sh script (#4783)

Georgi Gerganov %!s(int64=2) %!d(string=hai) anos
pai
achega
96e80dabc6
Modificáronse 1 ficheiros con 10 adicións e 5 borrados
  1. 10 5
      examples/base-translate.sh

+ 10 - 5
examples/base-translate.sh

@@ -8,14 +8,19 @@
 #   cd llama.cpp
 #   cd llama.cpp
 #   make -j
 #   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
   exit 1
 fi
 fi
 
 
+eargs=""
+if [ $# -gt 2 ]; then
+  eargs="${@:3}"
+fi
+
 ftmp="__llama.cpp_example_tmp__.txt"
 ftmp="__llama.cpp_example_tmp__.txt"
 trap "rm -f $ftmp" EXIT
 trap "rm -f $ftmp" EXIT
 
 
@@ -52,5 +57,5 @@ echo "$2
 
 
 model=$1
 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