diff --git a/llama-cpp-python/docker-entrypoint.sh b/llama-cpp-python/docker-entrypoint.sh index 38bea39..3b57ba3 100644 --- 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"