Просмотр исходного кода

examples : add -n to alpaca and gpt4all scripts (#706)

niansa/tuxifan 2 лет назад
Родитель
Сommit
107980d970
2 измененных файлов с 2 добавлено и 2 удалено
  1. 1 1
      examples/alpaca.sh
  2. 1 1
      examples/gpt4all.sh

+ 1 - 1
examples/alpaca.sh

@@ -7,4 +7,4 @@
 cd `dirname $0`
 cd `dirname $0`
 cd ..
 cd ..
 
 
-./main -m ./models/ggml-alpaca-7b-q4.bin --color -f ./prompts/alpaca.txt -ins -b 256 --top_k 10000 --temp 0.2 --repeat_penalty 1 -t 7
+./main -m ./models/ggml-alpaca-7b-q4.bin --color -f ./prompts/alpaca.txt --ctx_size 2048 -n -1 -ins -b 256 --top_k 10000 --temp 0.2 --repeat_penalty 1 -t 7

+ 1 - 1
examples/gpt4all.sh

@@ -10,6 +10,6 @@ cd ..
 ./main --color --instruct --threads 4 \
 ./main --color --instruct --threads 4 \
        --model ./models/gpt4all-7B/gpt4all-lora-quantized.bin \
        --model ./models/gpt4all-7B/gpt4all-lora-quantized.bin \
        --file ./prompts/alpaca.txt \
        --file ./prompts/alpaca.txt \
-       --batch_size 8 --ctx_size 2048 \
+       --batch_size 8 --ctx_size 2048 -n -1 \
        --repeat_last_n 64 --repeat_penalty 1.3 \
        --repeat_last_n 64 --repeat_penalty 1.3 \
        --n_predict 128 --temp 0.1 --top_k 40 --top_p 0.95
        --n_predict 128 --temp 0.1 --top_k 40 --top_p 0.95