|
@@ -20,12 +20,12 @@ if test "$WEBUI_SECRET_KEY $WEBUI_JWT_SECRET_KEY" = " "; then
|
|
|
WEBUI_SECRET_KEY=$(cat "$KEY_FILE")
|
|
|
fi
|
|
|
|
|
|
-if [ "$USE_OLLAMA_DOCKER" = "true" ]; then
|
|
|
+if [[ "${USE_OLLAMA_DOCKER,,}" == "true" ]]; then
|
|
|
echo "USE_OLLAMA is set to true, starting ollama serve."
|
|
|
ollama serve &
|
|
|
fi
|
|
|
|
|
|
-if [ "$USE_CUDA_DOCKER" = "true" ]; then
|
|
|
+if [[ "${USE_CUDA_DOCKER,,}" == "true" ]]; then
|
|
|
echo "CUDA is enabled, appending LD_LIBRARY_PATH to include torch/cudnn & cublas libraries."
|
|
|
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/lib/python3.11/site-packages/torch/lib:/usr/local/lib/python3.11/site-packages/nvidia/cudnn/lib"
|
|
|
fi
|