Fix scripts to support cross-platform execution

This commit is contained in:
Aizaixyq 2023-03-19 17:07:19 +08:00
parent d7def1a752
commit a8f0e40e30

View file

@ -7,8 +7,22 @@ if ! [[ "$1" =~ ^[0-9]{1,2}B$ ]]; then
exit 1 exit 1
fi fi
# Determine the binary file extension based on the platform
case "$(uname -s)" in
Linux*)
if grep -qE "(Microsoft|WSL)" /proc/version &> /dev/null ; then
ext=".exe" # WSL environment
else
ext=""
fi
;;
Darwin*) ext="";;
CYGWIN*|MINGW32*|MSYS*|MINGW*) ext=".exe";;
*) echo "Unknown platform"; exit 1;;
esac
for i in `ls models/$1/ggml-model-f16.bin*`; do for i in `ls models/$1/ggml-model-f16.bin*`; do
./quantize "$i" "${i/f16/q4_0}" 2 ./quantize${ext} "$i" "${i/f16/q4_0}" 2
if [[ "$2" == "--remove-f16" ]]; then if [[ "$2" == "--remove-f16" ]]; then
rm "$i" rm "$i"
fi fi