From 685d37418f741606c5ab9e8d664d06c6ada5d15a Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Mon, 29 Jan 2024 15:38:56 -0800
Subject: [PATCH] fix: run-ollama-docker.sh

---
 run-ollama-docker.sh | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/run-ollama-docker.sh b/run-ollama-docker.sh
index c8ce166..c2a025b 100644
--- a/run-ollama-docker.sh
+++ b/run-ollama-docker.sh
@@ -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