Unverified Commit b5c10ff1 authored by Timothy Jaeryang Baek's avatar Timothy Jaeryang Baek Committed by GitHub
Browse files

Merge pull request #598 from ollama-webui/run-ollama-sh-fix

fix: run-ollama-docker.sh
parents ee675e79 685d3741
......@@ -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
docker image prune -f
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment