Update docker-entrypoint.sh

3x3cut0r · Dec 7, 2023 · 679df56 · 679df56
1 parent 9f867db
commit 679df56
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llama-cpp-python/docker-entrypoint.sh b/llama-cpp-python/docker-entrypoint.sh
@@ -23,7 +23,7 @@ ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
 ############################
 
 # download default model
-if [ "$DOWNLOAD_DEFAULT_MODEL" = "True" ] || [ "$DOWNLOAD_DEFAULT_MODEL" = "true" ]; then
+if [ "${DOWNLOAD_DEFAULT_MODEL,,}" = "true" ]; then
     # download DEFAULT_MODEL only, if not exist
     if [ ! -e "/model/Llama-2-7b-Chat-GGUF/${DEFAULT_MODEL}" ]; then
         echo -e "\nINFO: start downloading default model:\nhttps://huggingface.co/TheBloke/Llama-2-7b-Chat-GGUF/resolve/main/${DEFAULT_MODEL}\n"
@@ -100,4 +100,4 @@ else
     fi
 fi
 
-exec runsvdir /runit-services
+exec runsvdir /runit-services