Update docker-entrypoint.sh

3x3cut0r · Dec 7, 2023 · b926d3b · b926d3b
1 parent 679df56
commit b926d3b
Showing 1 changed file with 1 addition and 1 deletion.
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" ]; then
+if [ "$DOWNLOAD_DEFAULT_MODEL" = "True" ] || [ "$DOWNLOAD_DEFAULT_MODEL" = "true" ] || [ "$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"