dnrops
/
ollama-open-webui
mirror of https://github.com/aileague/ollama-open-webui
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: run-ollama-docker.sh

This commit is contained in:
Timothy J. Baek 2024-01-29 15:38:56 -08:00
parent ee675e794f
commit 685d37418f
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
run-ollama-docker.sh
View File

@ -10,10 +10,10 @@ docker pull ollama/ollama:latest
docker_args="-d -v ollama:/root/.ollama -p $host_port:$container_port --name ollama ollama/ollama"
if [ "$use_gpu" == "y" ]; then
docker_args+=" --gpus=all"
if [ "$use_gpu" = "y" ]; then
docker_args="--gpus=all $docker_args"
fi
docker run "$docker_args"
docker run $docker_args
docker image prune -f