run.sh 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712
  1. #/bin/bash
  2. #
  3. # sample usage:
  4. #
  5. # mkdir tmp
  6. #
  7. # # CPU-only build
  8. # bash ./ci/run.sh ./tmp/results ./tmp/mnt
  9. #
  10. # # with CUDA support
  11. # GG_BUILD_CUDA=1 bash ./ci/run.sh ./tmp/results ./tmp/mnt
  12. #
  13. # # with SYCL support
  14. # GG_BUILD_SYCL=1 bash ./ci/run.sh ./tmp/results ./tmp/mnt
  15. #
  16. if [ -z "$2" ]; then
  17. echo "usage: $0 <output-dir> <mnt-dir>"
  18. exit 1
  19. fi
  20. mkdir -p "$1"
  21. mkdir -p "$2"
  22. OUT=$(realpath "$1")
  23. MNT=$(realpath "$2")
  24. rm -f "$OUT/*.log"
  25. rm -f "$OUT/*.exit"
  26. rm -f "$OUT/*.md"
  27. sd=`dirname $0`
  28. cd $sd/../
  29. SRC=`pwd`
  30. CMAKE_EXTRA="-DLLAMA_FATAL_WARNINGS=ON"
  31. if [ ! -z ${GG_BUILD_METAL} ]; then
  32. CMAKE_EXTRA="${CMAKE_EXTRA} -DLLAMA_METAL_SHADER_DEBUG=ON"
  33. fi
  34. if [ ! -z ${GG_BUILD_CUDA} ]; then
  35. CMAKE_EXTRA="${CMAKE_EXTRA} -DLLAMA_CUDA=1"
  36. fi
  37. if [ ! -z ${GG_BUILD_SYCL} ]; then
  38. if [ -z ${ONEAPI_ROOT} ]; then
  39. echo "Not detected ONEAPI_ROOT, please install oneAPI base toolkit and enable it by:"
  40. echo "source /opt/intel/oneapi/setvars.sh"
  41. exit 1
  42. fi
  43. CMAKE_EXTRA="${CMAKE_EXTRA} -DLLAMA_SYCL=1 DCMAKE_C_COMPILER=icx -DCMAKE_CXX_COMPILER=icpx -DLLAMA_SYCL_F16=ON"
  44. fi
  45. ## helpers
  46. # download a file if it does not exist or if it is outdated
  47. function gg_wget {
  48. local out=$1
  49. local url=$2
  50. local cwd=`pwd`
  51. mkdir -p $out
  52. cd $out
  53. # should not re-download if file is the same
  54. wget -nv -N $url
  55. cd $cwd
  56. }
  57. function gg_printf {
  58. printf -- "$@" >> $OUT/README.md
  59. }
  60. function gg_run {
  61. ci=$1
  62. set -o pipefail
  63. set -x
  64. gg_run_$ci | tee $OUT/$ci.log
  65. cur=$?
  66. echo "$cur" > $OUT/$ci.exit
  67. set +x
  68. set +o pipefail
  69. gg_sum_$ci
  70. ret=$((ret | cur))
  71. }
  72. ## ci
  73. # ctest_debug
  74. function gg_run_ctest_debug {
  75. cd ${SRC}
  76. rm -rf build-ci-debug && mkdir build-ci-debug && cd build-ci-debug
  77. set -e
  78. (time cmake -DCMAKE_BUILD_TYPE=Debug ${CMAKE_EXTRA} .. ) 2>&1 | tee -a $OUT/${ci}-cmake.log
  79. (time make -j ) 2>&1 | tee -a $OUT/${ci}-make.log
  80. (time ctest --output-on-failure -L main -E test-opt ) 2>&1 | tee -a $OUT/${ci}-ctest.log
  81. set +e
  82. }
  83. function gg_sum_ctest_debug {
  84. gg_printf '### %s\n\n' "${ci}"
  85. gg_printf 'Runs ctest in debug mode\n'
  86. gg_printf '- status: %s\n' "$(cat $OUT/${ci}.exit)"
  87. gg_printf '```\n'
  88. gg_printf '%s\n' "$(cat $OUT/${ci}-ctest.log)"
  89. gg_printf '```\n'
  90. gg_printf '\n'
  91. }
  92. # ctest_release
  93. function gg_run_ctest_release {
  94. cd ${SRC}
  95. rm -rf build-ci-release && mkdir build-ci-release && cd build-ci-release
  96. set -e
  97. (time cmake -DCMAKE_BUILD_TYPE=Release ${CMAKE_EXTRA} .. ) 2>&1 | tee -a $OUT/${ci}-cmake.log
  98. (time make -j ) 2>&1 | tee -a $OUT/${ci}-make.log
  99. if [ -z ${GG_BUILD_LOW_PERF} ]; then
  100. (time ctest --output-on-failure -L main ) 2>&1 | tee -a $OUT/${ci}-ctest.log
  101. else
  102. (time ctest --output-on-failure -L main -E test-opt ) 2>&1 | tee -a $OUT/${ci}-ctest.log
  103. fi
  104. set +e
  105. }
  106. function gg_sum_ctest_release {
  107. gg_printf '### %s\n\n' "${ci}"
  108. gg_printf 'Runs ctest in release mode\n'
  109. gg_printf '- status: %s\n' "$(cat $OUT/${ci}.exit)"
  110. gg_printf '```\n'
  111. gg_printf '%s\n' "$(cat $OUT/${ci}-ctest.log)"
  112. gg_printf '```\n'
  113. }
  114. # test_scripts_debug
  115. function gg_run_test_scripts_debug {
  116. cd ${SRC}
  117. set -e
  118. # TODO: too slow, run on dedicated node
  119. #(cd ./examples/gguf-split && time bash tests.sh "$SRC/build-ci-debug/bin" "$MNT/models") 2>&1 | tee -a $OUT/${ci}-scripts.log
  120. #(cd ./examples/quantize && time bash tests.sh "$SRC/build-ci-debug/bin" "$MNT/models") 2>&1 | tee -a $OUT/${ci}-scripts.log
  121. set +e
  122. }
  123. function gg_sum_test_scripts_debug {
  124. gg_printf '### %s\n\n' "${ci}"
  125. gg_printf 'Runs test scripts in debug mode\n'
  126. gg_printf '- status: %s\n' "$(cat $OUT/${ci}.exit)"
  127. gg_printf '```\n'
  128. gg_printf '%s\n' "$(cat $OUT/${ci}-scripts.log)"
  129. gg_printf '```\n'
  130. gg_printf '\n'
  131. }
  132. # test_scripts_release
  133. function gg_run_test_scripts_release {
  134. cd ${SRC}
  135. set -e
  136. (cd ./examples/gguf-split && time bash tests.sh "$SRC/build-ci-release/bin" "$MNT/models") 2>&1 | tee -a $OUT/${ci}-scripts.log
  137. (cd ./examples/quantize && time bash tests.sh "$SRC/build-ci-release/bin" "$MNT/models") 2>&1 | tee -a $OUT/${ci}-scripts.log
  138. set +e
  139. }
  140. function gg_sum_test_scripts_release {
  141. gg_printf '### %s\n\n' "${ci}"
  142. gg_printf 'Runs test scripts in release mode\n'
  143. gg_printf '- status: %s\n' "$(cat $OUT/${ci}.exit)"
  144. gg_printf '```\n'
  145. gg_printf '%s\n' "$(cat $OUT/${ci}-scripts.log)"
  146. gg_printf '```\n'
  147. gg_printf '\n'
  148. }
  149. function gg_get_model {
  150. local gguf_3b="$MNT/models/open-llama/3B-v2/ggml-model-f16.gguf"
  151. local gguf_7b="$MNT/models/open-llama/7B-v2/ggml-model-f16.gguf"
  152. if [[ -s $gguf_3b ]]; then
  153. echo -n "$gguf_3b"
  154. elif [[ -s $gguf_7b ]]; then
  155. echo -n "$gguf_7b"
  156. else
  157. echo >&2 "No model found. Can't run gg_run_ctest_with_model."
  158. exit 1
  159. fi
  160. }
  161. function gg_run_ctest_with_model_debug {
  162. cd ${SRC}
  163. local model; model=$(gg_get_model)
  164. cd build-ci-debug
  165. set -e
  166. (LLAMACPP_TEST_MODELFILE="$model" time ctest --output-on-failure -L model) 2>&1 | tee -a $OUT/${ci}-ctest.log
  167. set +e
  168. cd ..
  169. }
  170. function gg_run_ctest_with_model_release {
  171. cd ${SRC}
  172. local model; model=$(gg_get_model)
  173. cd build-ci-release
  174. set -e
  175. (LLAMACPP_TEST_MODELFILE="$model" time ctest --output-on-failure -L model) 2>&1 | tee -a $OUT/${ci}-ctest.log
  176. set +e
  177. cd ..
  178. }
  179. function gg_sum_ctest_with_model_debug {
  180. gg_printf '### %s\n\n' "${ci}"
  181. gg_printf 'Runs ctest with model files in debug mode\n'
  182. gg_printf '- status: %s\n' "$(cat $OUT/${ci}.exit)"
  183. gg_printf '```\n'
  184. gg_printf '%s\n' "$(cat $OUT/${ci}-ctest.log)"
  185. gg_printf '```\n'
  186. }
  187. function gg_sum_ctest_with_model_release {
  188. gg_printf '### %s\n\n' "${ci}"
  189. gg_printf 'Runs ctest with model files in release mode\n'
  190. gg_printf '- status: %s\n' "$(cat $OUT/${ci}.exit)"
  191. gg_printf '```\n'
  192. gg_printf '%s\n' "$(cat $OUT/${ci}-ctest.log)"
  193. gg_printf '```\n'
  194. }
  195. # open_llama_3b_v2
  196. function gg_run_open_llama_3b_v2 {
  197. cd ${SRC}
  198. gg_wget models-mnt/open-llama/3B-v2/ https://huggingface.co/openlm-research/open_llama_3b_v2/raw/main/config.json
  199. gg_wget models-mnt/open-llama/3B-v2/ https://huggingface.co/openlm-research/open_llama_3b_v2/resolve/main/tokenizer.model
  200. gg_wget models-mnt/open-llama/3B-v2/ https://huggingface.co/openlm-research/open_llama_3b_v2/raw/main/tokenizer_config.json
  201. gg_wget models-mnt/open-llama/3B-v2/ https://huggingface.co/openlm-research/open_llama_3b_v2/raw/main/special_tokens_map.json
  202. gg_wget models-mnt/open-llama/3B-v2/ https://huggingface.co/openlm-research/open_llama_3b_v2/resolve/main/pytorch_model.bin
  203. gg_wget models-mnt/open-llama/3B-v2/ https://huggingface.co/openlm-research/open_llama_3b_v2/raw/main/generation_config.json
  204. gg_wget models-mnt/wikitext/ https://huggingface.co/datasets/ggml-org/ci/resolve/main/wikitext-2-raw-v1.zip
  205. unzip -o models-mnt/wikitext/wikitext-2-raw-v1.zip -d models-mnt/wikitext/
  206. head -n 60 models-mnt/wikitext/wikitext-2-raw/wiki.test.raw > models-mnt/wikitext/wikitext-2-raw/wiki.test-60.raw
  207. path_models="../models-mnt/open-llama/3B-v2"
  208. path_wiki="../models-mnt/wikitext/wikitext-2-raw"
  209. rm -rf build-ci-release && mkdir build-ci-release && cd build-ci-release
  210. set -e
  211. (time cmake -DCMAKE_BUILD_TYPE=Release ${CMAKE_EXTRA} -DLLAMA_QKK_64=1 .. ) 2>&1 | tee -a $OUT/${ci}-cmake.log
  212. (time make -j ) 2>&1 | tee -a $OUT/${ci}-make.log
  213. python3 ../convert.py ${path_models}
  214. model_f16="${path_models}/ggml-model-f16.gguf"
  215. model_q8_0="${path_models}/ggml-model-q8_0.gguf"
  216. model_q4_0="${path_models}/ggml-model-q4_0.gguf"
  217. model_q4_1="${path_models}/ggml-model-q4_1.gguf"
  218. model_q5_0="${path_models}/ggml-model-q5_0.gguf"
  219. model_q5_1="${path_models}/ggml-model-q5_1.gguf"
  220. model_q2_k="${path_models}/ggml-model-q2_k.gguf"
  221. model_q3_k="${path_models}/ggml-model-q3_k.gguf"
  222. model_q4_k="${path_models}/ggml-model-q4_k.gguf"
  223. model_q5_k="${path_models}/ggml-model-q5_k.gguf"
  224. model_q6_k="${path_models}/ggml-model-q6_k.gguf"
  225. wiki_test_60="${path_wiki}/wiki.test-60.raw"
  226. ./bin/quantize ${model_f16} ${model_q8_0} q8_0
  227. ./bin/quantize ${model_f16} ${model_q4_0} q4_0
  228. ./bin/quantize ${model_f16} ${model_q4_1} q4_1
  229. ./bin/quantize ${model_f16} ${model_q5_0} q5_0
  230. ./bin/quantize ${model_f16} ${model_q5_1} q5_1
  231. ./bin/quantize ${model_f16} ${model_q2_k} q2_k
  232. ./bin/quantize ${model_f16} ${model_q3_k} q3_k
  233. ./bin/quantize ${model_f16} ${model_q4_k} q4_k
  234. ./bin/quantize ${model_f16} ${model_q5_k} q5_k
  235. ./bin/quantize ${model_f16} ${model_q6_k} q6_k
  236. (time ./bin/main --model ${model_f16} -s 1234 -n 64 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-f16.log
  237. (time ./bin/main --model ${model_q8_0} -s 1234 -n 64 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q8_0.log
  238. (time ./bin/main --model ${model_q4_0} -s 1234 -n 64 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q4_0.log
  239. (time ./bin/main --model ${model_q4_1} -s 1234 -n 64 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q4_1.log
  240. (time ./bin/main --model ${model_q5_0} -s 1234 -n 64 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q5_0.log
  241. (time ./bin/main --model ${model_q5_1} -s 1234 -n 64 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q5_1.log
  242. (time ./bin/main --model ${model_q2_k} -s 1234 -n 64 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q2_k.log
  243. (time ./bin/main --model ${model_q3_k} -s 1234 -n 64 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q3_k.log
  244. (time ./bin/main --model ${model_q4_k} -s 1234 -n 64 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q4_k.log
  245. (time ./bin/main --model ${model_q5_k} -s 1234 -n 64 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q5_k.log
  246. (time ./bin/main --model ${model_q6_k} -s 1234 -n 64 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q6_k.log
  247. (time ./bin/perplexity --model ${model_f16} -f ${wiki_test_60} -c 128 -b 128 --chunks 1 ) 2>&1 | tee -a $OUT/${ci}-tg-f16.log
  248. (time ./bin/perplexity --model ${model_q8_0} -f ${wiki_test_60} -c 128 -b 128 --chunks 1 ) 2>&1 | tee -a $OUT/${ci}-tg-q8_0.log
  249. (time ./bin/perplexity --model ${model_q4_0} -f ${wiki_test_60} -c 128 -b 128 --chunks 1 ) 2>&1 | tee -a $OUT/${ci}-tg-q4_0.log
  250. (time ./bin/perplexity --model ${model_q4_1} -f ${wiki_test_60} -c 128 -b 128 --chunks 1 ) 2>&1 | tee -a $OUT/${ci}-tg-q4_1.log
  251. (time ./bin/perplexity --model ${model_q5_0} -f ${wiki_test_60} -c 128 -b 128 --chunks 1 ) 2>&1 | tee -a $OUT/${ci}-tg-q5_0.log
  252. (time ./bin/perplexity --model ${model_q5_1} -f ${wiki_test_60} -c 128 -b 128 --chunks 1 ) 2>&1 | tee -a $OUT/${ci}-tg-q5_1.log
  253. (time ./bin/perplexity --model ${model_q2_k} -f ${wiki_test_60} -c 128 -b 128 --chunks 1 ) 2>&1 | tee -a $OUT/${ci}-tg-q2_k.log
  254. (time ./bin/perplexity --model ${model_q3_k} -f ${wiki_test_60} -c 128 -b 128 --chunks 1 ) 2>&1 | tee -a $OUT/${ci}-tg-q3_k.log
  255. (time ./bin/perplexity --model ${model_q4_k} -f ${wiki_test_60} -c 128 -b 128 --chunks 1 ) 2>&1 | tee -a $OUT/${ci}-tg-q4_k.log
  256. (time ./bin/perplexity --model ${model_q5_k} -f ${wiki_test_60} -c 128 -b 128 --chunks 1 ) 2>&1 | tee -a $OUT/${ci}-tg-q5_k.log
  257. (time ./bin/perplexity --model ${model_q6_k} -f ${wiki_test_60} -c 128 -b 128 --chunks 1 ) 2>&1 | tee -a $OUT/${ci}-tg-q6_k.log
  258. (time ./bin/imatrix --model ${model_f16} -f ${wiki_test_60} -c 128 -b 128 --chunks 1 ) 2>&1 | tee -a $OUT/${ci}-imatrix.log
  259. (time ./bin/save-load-state --model ${model_q4_0} ) 2>&1 | tee -a $OUT/${ci}-save-load-state.log
  260. (time ./bin/save-load-state -fa --model ${model_q4_0} ) 2>&1 | tee -a $OUT/${ci}-save-load-state.log
  261. function check_ppl {
  262. qnt="$1"
  263. ppl=$(echo "$2" | grep -oE "[0-9]+\.[0-9]+" | tail -n 1)
  264. if [ $(echo "$ppl > 20.0" | bc) -eq 1 ]; then
  265. printf ' - %s @ %s (FAIL: ppl > 20.0)\n' "$qnt" "$ppl"
  266. return 20
  267. fi
  268. printf ' - %s @ %s OK\n' "$qnt" "$ppl"
  269. return 0
  270. }
  271. check_ppl "f16" "$(cat $OUT/${ci}-tg-f16.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  272. check_ppl "q8_0" "$(cat $OUT/${ci}-tg-q8_0.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  273. check_ppl "q4_0" "$(cat $OUT/${ci}-tg-q4_0.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  274. check_ppl "q4_1" "$(cat $OUT/${ci}-tg-q4_1.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  275. check_ppl "q5_0" "$(cat $OUT/${ci}-tg-q5_0.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  276. check_ppl "q5_1" "$(cat $OUT/${ci}-tg-q5_1.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  277. check_ppl "q2_k" "$(cat $OUT/${ci}-tg-q2_k.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  278. check_ppl "q3_k" "$(cat $OUT/${ci}-tg-q3_k.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  279. check_ppl "q4_k" "$(cat $OUT/${ci}-tg-q4_k.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  280. check_ppl "q5_k" "$(cat $OUT/${ci}-tg-q5_k.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  281. check_ppl "q6_k" "$(cat $OUT/${ci}-tg-q6_k.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  282. cat $OUT/${ci}-imatrix.log | grep "Final" >> $OUT/${ci}-imatrix-sum.log
  283. # lora
  284. function compare_ppl {
  285. qnt="$1"
  286. ppl1=$(echo "$2" | grep -oE "[0-9]+\.[0-9]+" | tail -n 1)
  287. ppl2=$(echo "$3" | grep -oE "[0-9]+\.[0-9]+" | tail -n 1)
  288. if [ $(echo "$ppl1 < $ppl2" | bc) -eq 1 ]; then
  289. printf ' - %s @ %s (FAIL: %s > %s)\n' "$qnt" "$ppl" "$ppl1" "$ppl2"
  290. return 20
  291. fi
  292. printf ' - %s @ %s %s OK\n' "$qnt" "$ppl1" "$ppl2"
  293. return 0
  294. }
  295. path_lora="../models-mnt/open-llama/3B-v2/lora"
  296. path_shakespeare="../models-mnt/shakespeare"
  297. shakespeare="${path_shakespeare}/shakespeare.txt"
  298. lora_shakespeare="${path_lora}/ggml-adapter-model.bin"
  299. gg_wget ${path_lora} https://huggingface.co/slaren/open_llama_3b_v2_shakespeare_lora/resolve/main/adapter_config.json
  300. gg_wget ${path_lora} https://huggingface.co/slaren/open_llama_3b_v2_shakespeare_lora/resolve/main/adapter_model.bin
  301. gg_wget ${path_shakespeare} https://huggingface.co/slaren/open_llama_3b_v2_shakespeare_lora/resolve/main/shakespeare.txt
  302. python3 ../convert-lora-to-ggml.py ${path_lora}
  303. # f16
  304. (time ./bin/perplexity --model ${model_f16} -f ${shakespeare} -c 128 -b 128 --chunks 1 ) 2>&1 | tee -a $OUT/${ci}-ppl-shakespeare-f16.log
  305. (time ./bin/perplexity --model ${model_f16} -f ${shakespeare} --lora ${lora_shakespeare} -c 128 -b 128 --chunks 1 ) 2>&1 | tee -a $OUT/${ci}-ppl-shakespeare-lora-f16.log
  306. compare_ppl "f16 shakespeare" "$(cat $OUT/${ci}-ppl-shakespeare-f16.log | grep "^\[1\]")" "$(cat $OUT/${ci}-ppl-shakespeare-lora-f16.log | grep "^\[1\]")" | tee -a $OUT/${ci}-lora-ppl.log
  307. # q8_0
  308. (time ./bin/perplexity --model ${model_q8_0} -f ${shakespeare} -c 128 -b 128 --chunks 1 ) 2>&1 | tee -a $OUT/${ci}-ppl-shakespeare-q8_0.log
  309. (time ./bin/perplexity --model ${model_q8_0} -f ${shakespeare} --lora ${lora_shakespeare} -c 128 -b 128 --chunks 1 ) 2>&1 | tee -a $OUT/${ci}-ppl-shakespeare-lora-q8_0.log
  310. compare_ppl "q8_0 shakespeare" "$(cat $OUT/${ci}-ppl-shakespeare-q8_0.log | grep "^\[1\]")" "$(cat $OUT/${ci}-ppl-shakespeare-lora-q8_0.log | grep "^\[1\]")" | tee -a $OUT/${ci}-lora-ppl.log
  311. # q8_0 + f16 lora-base
  312. (time ./bin/perplexity --model ${model_q8_0} -f ${shakespeare} --lora ${lora_shakespeare} --lora-base ${model_f16} -c 128 -b 128 --chunks 1 ) 2>&1 | tee -a $OUT/${ci}-ppl-shakespeare-lora-q8_0-f16.log
  313. compare_ppl "q8_0 / f16 base shakespeare" "$(cat $OUT/${ci}-ppl-shakespeare-q8_0.log | grep "^\[1\]")" "$(cat $OUT/${ci}-ppl-shakespeare-lora-q8_0-f16.log | grep "^\[1\]")" | tee -a $OUT/${ci}-lora-ppl.log
  314. set +e
  315. }
  316. function gg_sum_open_llama_3b_v2 {
  317. gg_printf '### %s\n\n' "${ci}"
  318. gg_printf 'OpenLLaMA 3B-v2:\n'
  319. gg_printf '- status: %s\n' "$(cat $OUT/${ci}.exit)"
  320. gg_printf '- perplexity:\n%s\n' "$(cat $OUT/${ci}-ppl.log)"
  321. gg_printf '- imatrix:\n```\n%s\n```\n' "$(cat $OUT/${ci}-imatrix-sum.log)"
  322. gg_printf '- lora:\n%s\n' "$(cat $OUT/${ci}-lora-ppl.log)"
  323. gg_printf '- f16: \n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-f16.log)"
  324. gg_printf '- q8_0:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q8_0.log)"
  325. gg_printf '- q4_0:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q4_0.log)"
  326. gg_printf '- q4_1:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q4_1.log)"
  327. gg_printf '- q5_0:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q5_0.log)"
  328. gg_printf '- q5_1:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q5_1.log)"
  329. gg_printf '- q2_k:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q2_k.log)"
  330. gg_printf '- q3_k:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q3_k.log)"
  331. gg_printf '- q4_k:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q4_k.log)"
  332. gg_printf '- q5_k:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q5_k.log)"
  333. gg_printf '- q6_k:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q6_k.log)"
  334. gg_printf '- save-load-state: \n```\n%s\n```\n' "$(cat $OUT/${ci}-save-load-state.log)"
  335. gg_printf '- shakespeare (f16):\n```\n%s\n```\n' "$(cat $OUT/${ci}-ppl-shakespeare-f16.log)"
  336. gg_printf '- shakespeare (f16 lora):\n```\n%s\n```\n' "$(cat $OUT/${ci}-ppl-shakespeare-lora-f16.log)"
  337. gg_printf '- shakespeare (q8_0):\n```\n%s\n```\n' "$(cat $OUT/${ci}-ppl-shakespeare-q8_0.log)"
  338. gg_printf '- shakespeare (q8_0 lora):\n```\n%s\n```\n' "$(cat $OUT/${ci}-ppl-shakespeare-lora-q8_0.log)"
  339. gg_printf '- shakespeare (q8_0 / f16 base lora):\n```\n%s\n```\n' "$(cat $OUT/${ci}-ppl-shakespeare-lora-q8_0-f16.log)"
  340. }
  341. # open_llama_7b_v2
  342. # requires: GG_BUILD_CUDA
  343. function gg_run_open_llama_7b_v2 {
  344. cd ${SRC}
  345. gg_wget models-mnt/open-llama/7B-v2/ https://huggingface.co/openlm-research/open_llama_7b_v2/raw/main/config.json
  346. gg_wget models-mnt/open-llama/7B-v2/ https://huggingface.co/openlm-research/open_llama_7b_v2/resolve/main/tokenizer.model
  347. gg_wget models-mnt/open-llama/7B-v2/ https://huggingface.co/openlm-research/open_llama_7b_v2/raw/main/tokenizer_config.json
  348. gg_wget models-mnt/open-llama/7B-v2/ https://huggingface.co/openlm-research/open_llama_7b_v2/raw/main/special_tokens_map.json
  349. gg_wget models-mnt/open-llama/7B-v2/ https://huggingface.co/openlm-research/open_llama_7b_v2/raw/main/pytorch_model.bin.index.json
  350. gg_wget models-mnt/open-llama/7B-v2/ https://huggingface.co/openlm-research/open_llama_7b_v2/resolve/main/pytorch_model-00001-of-00002.bin
  351. gg_wget models-mnt/open-llama/7B-v2/ https://huggingface.co/openlm-research/open_llama_7b_v2/resolve/main/pytorch_model-00002-of-00002.bin
  352. gg_wget models-mnt/open-llama/7B-v2/ https://huggingface.co/openlm-research/open_llama_7b_v2/raw/main/generation_config.json
  353. gg_wget models-mnt/wikitext/ https://huggingface.co/datasets/ggml-org/ci/resolve/main/wikitext-2-raw-v1.zip
  354. unzip -o models-mnt/wikitext/wikitext-2-raw-v1.zip -d models-mnt/wikitext/
  355. path_models="../models-mnt/open-llama/7B-v2"
  356. path_wiki="../models-mnt/wikitext/wikitext-2-raw"
  357. rm -rf build-ci-release && mkdir build-ci-release && cd build-ci-release
  358. set -e
  359. (time cmake -DCMAKE_BUILD_TYPE=Release ${CMAKE_EXTRA} -DLLAMA_CUDA=1 .. ) 2>&1 | tee -a $OUT/${ci}-cmake.log
  360. (time make -j ) 2>&1 | tee -a $OUT/${ci}-make.log
  361. python3 ../convert.py ${path_models}
  362. model_f16="${path_models}/ggml-model-f16.gguf"
  363. model_q8_0="${path_models}/ggml-model-q8_0.gguf"
  364. model_q4_0="${path_models}/ggml-model-q4_0.gguf"
  365. model_q4_1="${path_models}/ggml-model-q4_1.gguf"
  366. model_q5_0="${path_models}/ggml-model-q5_0.gguf"
  367. model_q5_1="${path_models}/ggml-model-q5_1.gguf"
  368. model_q2_k="${path_models}/ggml-model-q2_k.gguf"
  369. model_q3_k="${path_models}/ggml-model-q3_k.gguf"
  370. model_q4_k="${path_models}/ggml-model-q4_k.gguf"
  371. model_q5_k="${path_models}/ggml-model-q5_k.gguf"
  372. model_q6_k="${path_models}/ggml-model-q6_k.gguf"
  373. wiki_test="${path_wiki}/wiki.test.raw"
  374. ./bin/quantize ${model_f16} ${model_q8_0} q8_0
  375. ./bin/quantize ${model_f16} ${model_q4_0} q4_0
  376. ./bin/quantize ${model_f16} ${model_q4_1} q4_1
  377. ./bin/quantize ${model_f16} ${model_q5_0} q5_0
  378. ./bin/quantize ${model_f16} ${model_q5_1} q5_1
  379. ./bin/quantize ${model_f16} ${model_q2_k} q2_k
  380. ./bin/quantize ${model_f16} ${model_q3_k} q3_k
  381. ./bin/quantize ${model_f16} ${model_q4_k} q4_k
  382. ./bin/quantize ${model_f16} ${model_q5_k} q5_k
  383. ./bin/quantize ${model_f16} ${model_q6_k} q6_k
  384. (time ./bin/main --model ${model_f16} -t 1 -ngl 999 -s 1234 -n 256 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-f16.log
  385. (time ./bin/main --model ${model_q8_0} -t 1 -ngl 999 -s 1234 -n 256 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q8_0.log
  386. (time ./bin/main --model ${model_q4_0} -t 1 -ngl 999 -s 1234 -n 256 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q4_0.log
  387. (time ./bin/main --model ${model_q4_1} -t 1 -ngl 999 -s 1234 -n 256 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q4_1.log
  388. (time ./bin/main --model ${model_q5_0} -t 1 -ngl 999 -s 1234 -n 256 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q5_0.log
  389. (time ./bin/main --model ${model_q5_1} -t 1 -ngl 999 -s 1234 -n 256 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q5_1.log
  390. (time ./bin/main --model ${model_q2_k} -t 1 -ngl 999 -s 1234 -n 256 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q2_k.log
  391. (time ./bin/main --model ${model_q3_k} -t 1 -ngl 999 -s 1234 -n 256 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q3_k.log
  392. (time ./bin/main --model ${model_q4_k} -t 1 -ngl 999 -s 1234 -n 256 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q4_k.log
  393. (time ./bin/main --model ${model_q5_k} -t 1 -ngl 999 -s 1234 -n 256 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q5_k.log
  394. (time ./bin/main --model ${model_q6_k} -t 1 -ngl 999 -s 1234 -n 256 --ignore-eos -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q6_k.log
  395. (time ./bin/perplexity --model ${model_f16} -f ${wiki_test} -t 1 -ngl 999 -c 2048 -b 512 --chunks 4 ) 2>&1 | tee -a $OUT/${ci}-tg-f16.log
  396. (time ./bin/perplexity --model ${model_q8_0} -f ${wiki_test} -t 1 -ngl 999 -c 2048 -b 512 --chunks 4 ) 2>&1 | tee -a $OUT/${ci}-tg-q8_0.log
  397. (time ./bin/perplexity --model ${model_q4_0} -f ${wiki_test} -t 1 -ngl 999 -c 2048 -b 512 --chunks 4 ) 2>&1 | tee -a $OUT/${ci}-tg-q4_0.log
  398. (time ./bin/perplexity --model ${model_q4_1} -f ${wiki_test} -t 1 -ngl 999 -c 2048 -b 512 --chunks 4 ) 2>&1 | tee -a $OUT/${ci}-tg-q4_1.log
  399. (time ./bin/perplexity --model ${model_q5_0} -f ${wiki_test} -t 1 -ngl 999 -c 2048 -b 512 --chunks 4 ) 2>&1 | tee -a $OUT/${ci}-tg-q5_0.log
  400. (time ./bin/perplexity --model ${model_q5_1} -f ${wiki_test} -t 1 -ngl 999 -c 2048 -b 512 --chunks 4 ) 2>&1 | tee -a $OUT/${ci}-tg-q5_1.log
  401. (time ./bin/perplexity --model ${model_q2_k} -f ${wiki_test} -t 1 -ngl 999 -c 2048 -b 512 --chunks 4 ) 2>&1 | tee -a $OUT/${ci}-tg-q2_k.log
  402. (time ./bin/perplexity --model ${model_q3_k} -f ${wiki_test} -t 1 -ngl 999 -c 2048 -b 512 --chunks 4 ) 2>&1 | tee -a $OUT/${ci}-tg-q3_k.log
  403. (time ./bin/perplexity --model ${model_q4_k} -f ${wiki_test} -t 1 -ngl 999 -c 2048 -b 512 --chunks 4 ) 2>&1 | tee -a $OUT/${ci}-tg-q4_k.log
  404. (time ./bin/perplexity --model ${model_q5_k} -f ${wiki_test} -t 1 -ngl 999 -c 2048 -b 512 --chunks 4 ) 2>&1 | tee -a $OUT/${ci}-tg-q5_k.log
  405. (time ./bin/perplexity --model ${model_q6_k} -f ${wiki_test} -t 1 -ngl 999 -c 2048 -b 512 --chunks 4 ) 2>&1 | tee -a $OUT/${ci}-tg-q6_k.log
  406. (time ./bin/imatrix --model ${model_f16} -f ${wiki_test} -t 1 -ngl 999 -c 2048 -b 512 --chunks 4 ) 2>&1 | tee -a $OUT/${ci}-imatrix.log
  407. (time ./bin/save-load-state -ngl 10 --model ${model_q4_0} ) 2>&1 | tee -a $OUT/${ci}-save-load-state.log
  408. (time ./bin/save-load-state -fa -ngl 10 --model ${model_q4_0} ) 2>&1 | tee -a $OUT/${ci}-save-load-state.log
  409. (time ./bin/save-load-state -ngl 99 --model ${model_q4_0} ) 2>&1 | tee -a $OUT/${ci}-save-load-state.log
  410. (time ./bin/save-load-state -fa -ngl 99 --model ${model_q4_0} ) 2>&1 | tee -a $OUT/${ci}-save-load-state.log
  411. function check_ppl {
  412. qnt="$1"
  413. ppl=$(echo "$2" | grep -oE "[0-9]+\.[0-9]+" | tail -n 1)
  414. if [ $(echo "$ppl > 20.0" | bc) -eq 1 ]; then
  415. printf ' - %s @ %s (FAIL: ppl > 20.0)\n' "$qnt" "$ppl"
  416. return 20
  417. fi
  418. printf ' - %s @ %s OK\n' "$qnt" "$ppl"
  419. return 0
  420. }
  421. check_ppl "f16" "$(cat $OUT/${ci}-tg-f16.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  422. check_ppl "q8_0" "$(cat $OUT/${ci}-tg-q8_0.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  423. check_ppl "q4_0" "$(cat $OUT/${ci}-tg-q4_0.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  424. check_ppl "q4_1" "$(cat $OUT/${ci}-tg-q4_1.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  425. check_ppl "q5_0" "$(cat $OUT/${ci}-tg-q5_0.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  426. check_ppl "q5_1" "$(cat $OUT/${ci}-tg-q5_1.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  427. check_ppl "q2_k" "$(cat $OUT/${ci}-tg-q2_k.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  428. check_ppl "q3_k" "$(cat $OUT/${ci}-tg-q3_k.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  429. check_ppl "q4_k" "$(cat $OUT/${ci}-tg-q4_k.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  430. check_ppl "q5_k" "$(cat $OUT/${ci}-tg-q5_k.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  431. check_ppl "q6_k" "$(cat $OUT/${ci}-tg-q6_k.log | grep "^\[1\]")" | tee -a $OUT/${ci}-ppl.log
  432. cat $OUT/${ci}-imatrix.log | grep "Final" >> $OUT/${ci}-imatrix-sum.log
  433. # lora
  434. function compare_ppl {
  435. qnt="$1"
  436. ppl1=$(echo "$2" | grep -oE "[0-9]+\.[0-9]+" | tail -n 1)
  437. ppl2=$(echo "$3" | grep -oE "[0-9]+\.[0-9]+" | tail -n 1)
  438. if [ $(echo "$ppl1 < $ppl2" | bc) -eq 1 ]; then
  439. printf ' - %s @ %s (FAIL: %s > %s)\n' "$qnt" "$ppl" "$ppl1" "$ppl2"
  440. return 20
  441. fi
  442. printf ' - %s @ %s %s OK\n' "$qnt" "$ppl1" "$ppl2"
  443. return 0
  444. }
  445. path_lora="../models-mnt/open-llama/7B-v2/lora"
  446. path_shakespeare="../models-mnt/shakespeare"
  447. shakespeare="${path_shakespeare}/shakespeare.txt"
  448. lora_shakespeare="${path_lora}/ggml-adapter-model.bin"
  449. gg_wget ${path_lora} https://huggingface.co/slaren/open_llama_7b_v2_shakespeare_lora/resolve/main/adapter_config.json
  450. gg_wget ${path_lora} https://huggingface.co/slaren/open_llama_7b_v2_shakespeare_lora/resolve/main/adapter_model.bin
  451. gg_wget ${path_shakespeare} https://huggingface.co/slaren/open_llama_7b_v2_shakespeare_lora/resolve/main/shakespeare.txt
  452. python3 ../convert-lora-to-ggml.py ${path_lora}
  453. # f16
  454. (time ./bin/perplexity --model ${model_f16} -f ${shakespeare} -t 1 -ngl 999 -c 2048 -b 512 --chunks 3 ) 2>&1 | tee -a $OUT/${ci}-ppl-shakespeare-f16.log
  455. (time ./bin/perplexity --model ${model_f16} -f ${shakespeare} --lora ${lora_shakespeare} -t 1 -ngl 999 -c 2048 -b 512 --chunks 3 ) 2>&1 | tee -a $OUT/${ci}-ppl-shakespeare-lora-f16.log
  456. compare_ppl "f16 shakespeare" "$(cat $OUT/${ci}-ppl-shakespeare-f16.log | grep "^\[1\]")" "$(cat $OUT/${ci}-ppl-shakespeare-lora-f16.log | grep "^\[1\]")" | tee -a $OUT/${ci}-lora-ppl.log
  457. # currently not supported by the CUDA backend
  458. # q8_0
  459. #(time ./bin/perplexity --model ${model_q8_0} -f ${shakespeare} -t 1 -ngl 999 -c 2048 -b 512 --chunks 3 ) 2>&1 | tee -a $OUT/${ci}-ppl-shakespeare-q8_0.log
  460. #(time ./bin/perplexity --model ${model_q8_0} -f ${shakespeare} --lora ${lora_shakespeare} -t 1 -ngl 999 -c 2048 -b 512 --chunks 3 ) 2>&1 | tee -a $OUT/${ci}-ppl-shakespeare-lora-q8_0.log
  461. #compare_ppl "q8_0 shakespeare" "$(cat $OUT/${ci}-ppl-shakespeare-q8_0.log | grep "^\[1\]")" "$(cat $OUT/${ci}-ppl-shakespeare-lora-q8_0.log | grep "^\[1\]")" | tee -a $OUT/${ci}-lora-ppl.log
  462. # q8_0 + f16 lora-base
  463. #(time ./bin/perplexity --model ${model_q8_0} -f ${shakespeare} --lora ${lora_shakespeare} --lora-base ${model_f16} -t 1 -ngl 999 -c 2048 -b 512 --chunks 3 ) 2>&1 | tee -a $OUT/${ci}-ppl-shakespeare-lora-q8_0-f16.log
  464. #compare_ppl "q8_0 / f16 shakespeare" "$(cat $OUT/${ci}-ppl-shakespeare-q8_0.log | grep "^\[1\]")" "$(cat $OUT/${ci}-ppl-shakespeare-lora-q8_0-f16.log | grep "^\[1\]")" | tee -a $OUT/${ci}-lora-ppl.log
  465. set +e
  466. }
  467. function gg_sum_open_llama_7b_v2 {
  468. gg_printf '### %s\n\n' "${ci}"
  469. gg_printf 'OpenLLaMA 7B-v2:\n'
  470. gg_printf '- status: %s\n' "$(cat $OUT/${ci}.exit)"
  471. gg_printf '- perplexity:\n%s\n' "$(cat $OUT/${ci}-ppl.log)"
  472. gg_printf '- imatrix:\n```\n%s\n```\n' "$(cat $OUT/${ci}-imatrix-sum.log)"
  473. gg_printf '- lora:\n%s\n' "$(cat $OUT/${ci}-lora-ppl.log)"
  474. gg_printf '- f16: \n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-f16.log)"
  475. gg_printf '- q8_0:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q8_0.log)"
  476. gg_printf '- q4_0:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q4_0.log)"
  477. gg_printf '- q4_1:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q4_1.log)"
  478. gg_printf '- q5_0:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q5_0.log)"
  479. gg_printf '- q5_1:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q5_1.log)"
  480. gg_printf '- q2_k:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q2_k.log)"
  481. gg_printf '- q3_k:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q3_k.log)"
  482. gg_printf '- q4_k:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q4_k.log)"
  483. gg_printf '- q5_k:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q5_k.log)"
  484. gg_printf '- q6_k:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q6_k.log)"
  485. gg_printf '- save-load-state: \n```\n%s\n```\n' "$(cat $OUT/${ci}-save-load-state.log)"
  486. gg_printf '- shakespeare (f16):\n```\n%s\n```\n' "$(cat $OUT/${ci}-ppl-shakespeare-f16.log)"
  487. gg_printf '- shakespeare (f16 lora):\n```\n%s\n```\n' "$(cat $OUT/${ci}-ppl-shakespeare-lora-f16.log)"
  488. #gg_printf '- shakespeare (q8_0):\n```\n%s\n```\n' "$(cat $OUT/${ci}-ppl-shakespeare-q8_0.log)"
  489. #gg_printf '- shakespeare (q8_0 lora):\n```\n%s\n```\n' "$(cat $OUT/${ci}-ppl-shakespeare-lora-q8_0.log)"
  490. #gg_printf '- shakespeare (q8_0 / f16 base lora):\n```\n%s\n```\n' "$(cat $OUT/${ci}-ppl-shakespeare-lora-q8_0-f16.log)"
  491. }
  492. # bge-small
  493. function gg_run_embd_bge_small {
  494. cd ${SRC}
  495. gg_wget models-mnt/bge-small/ https://huggingface.co/BAAI/bge-small-en-v1.5/raw/main/config.json
  496. gg_wget models-mnt/bge-small/ https://huggingface.co/BAAI/bge-small-en-v1.5/raw/main/tokenizer.json
  497. gg_wget models-mnt/bge-small/ https://huggingface.co/BAAI/bge-small-en-v1.5/raw/main/tokenizer_config.json
  498. gg_wget models-mnt/bge-small/ https://huggingface.co/BAAI/bge-small-en-v1.5/raw/main/special_tokens_map.json
  499. gg_wget models-mnt/bge-small/ https://huggingface.co/BAAI/bge-small-en-v1.5/resolve/main/pytorch_model.bin
  500. gg_wget models-mnt/bge-small/ https://huggingface.co/BAAI/bge-small-en-v1.5/raw/main/sentence_bert_config.json
  501. gg_wget models-mnt/bge-small/ https://huggingface.co/BAAI/bge-small-en-v1.5/raw/main/vocab.txt
  502. gg_wget models-mnt/bge-small/ https://huggingface.co/BAAI/bge-small-en-v1.5/raw/main/modules.json
  503. gg_wget models-mnt/bge-small/ https://huggingface.co/BAAI/bge-small-en-v1.5/raw/main/config.json
  504. gg_wget models-mnt/bge-small/1_Pooling https://huggingface.co/BAAI/bge-small-en-v1.5/raw/main/1_Pooling/config.json
  505. path_models="../models-mnt/bge-small"
  506. rm -rf build-ci-release && mkdir build-ci-release && cd build-ci-release
  507. set -e
  508. (time cmake -DCMAKE_BUILD_TYPE=Release ${CMAKE_EXTRA} .. ) 2>&1 | tee -a $OUT/${ci}-cmake.log
  509. (time make -j ) 2>&1 | tee -a $OUT/${ci}-make.log
  510. python3 ../convert-hf-to-gguf.py ${path_models}
  511. model_f16="${path_models}/ggml-model-f16.gguf"
  512. model_q8_0="${path_models}/ggml-model-q8_0.gguf"
  513. ./bin/quantize ${model_f16} ${model_q8_0} q8_0
  514. (time ./bin/embedding --model ${model_f16} -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-f16.log
  515. (time ./bin/embedding --model ${model_q8_0} -p "I believe the meaning of life is" ) 2>&1 | tee -a $OUT/${ci}-tg-q8_0.log
  516. set +e
  517. }
  518. function gg_sum_embd_bge_small {
  519. gg_printf '### %s\n\n' "${ci}"
  520. gg_printf 'BGE Small (BERT):\n'
  521. gg_printf '- status: %s\n' "$(cat $OUT/${ci}.exit)"
  522. gg_printf '- f16: \n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-f16.log)"
  523. gg_printf '- q8_0:\n```\n%s\n```\n' "$(cat $OUT/${ci}-tg-q8_0.log)"
  524. }
  525. ## main
  526. if [ -z ${GG_BUILD_LOW_PERF} ]; then
  527. # Create symlink: ./llama.cpp/models-mnt -> $MNT/models/models-mnt
  528. rm -rf ${SRC}/models-mnt
  529. mnt_models=${MNT}/models
  530. mkdir -p ${mnt_models}
  531. ln -sfn ${mnt_models} ${SRC}/models-mnt
  532. # Create a fresh python3 venv and enter it
  533. python3 -m venv "$MNT/venv"
  534. source "$MNT/venv/bin/activate"
  535. pip install -r ${SRC}/requirements.txt --disable-pip-version-check
  536. pip install --editable gguf-py --disable-pip-version-check
  537. fi
  538. ret=0
  539. test $ret -eq 0 && gg_run ctest_debug
  540. test $ret -eq 0 && gg_run ctest_release
  541. if [ -z ${GG_BUILD_LOW_PERF} ]; then
  542. test $ret -eq 0 && gg_run embd_bge_small
  543. test $ret -eq 0 && gg_run test_scripts_debug
  544. test $ret -eq 0 && gg_run test_scripts_release
  545. if [ -z ${GG_BUILD_VRAM_GB} ] || [ ${GG_BUILD_VRAM_GB} -ge 8 ]; then
  546. if [ -z ${GG_BUILD_CUDA} ]; then
  547. test $ret -eq 0 && gg_run open_llama_3b_v2
  548. else
  549. test $ret -eq 0 && gg_run open_llama_7b_v2
  550. fi
  551. test $ret -eq 0 && gg_run ctest_with_model_debug
  552. test $ret -eq 0 && gg_run ctest_with_model_release
  553. fi
  554. fi
  555. exit $ret