From 156724a0549ad140dc7ed58b4b6f554afb303aaf Mon Sep 17 00:00:00 2001
From: Meenakshi Sharma <163925564+nvda-mesharma@users.noreply.github.com>
Date: Wed, 4 Sep 2024 16:08:05 -0700
Subject: [PATCH] Update build_docker.sh

---
 ci/build/build_docker.sh | 18 ++++++++----------
 1 file changed, 8 insertions(+), 10 deletions(-)

diff --git a/ci/build/build_docker.sh b/ci/build/build_docker.sh
index 34752a96..76ea2ed5 100755
--- a/ci/build/build_docker.sh
+++ b/ci/build/build_docker.sh
@@ -25,15 +25,13 @@
 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
-# Get latest VLLM RELEASED VERSION from https://github.com/triton-inference-server/vllm_backend/releases
-TAG=$(curl https://api.github.com/repos/triton-inference-server/vllm_backend/releases/latest | grep -i "tag_name" | awk -F '"' '{print $4}')
-export TRITON_CONTAINER_VERSION=${TAG#v} # example: 24.06
-if [ -z "$TRITON_CONTAINER_VERSION" ]
-            then
-              echo "\$TRITON_CONTAINER_VERSION is NULL, setting it to 24.08"
-              TRITON_CONTAINER_VERSION=24.08
-            else
-              echo "\$TRITON_CONTAINER_VERSION is NOT NULL"
-          fi
+while getopts t:v: flag
+do
+    case "${flag}" in
+        u) TRITON_CONTAINER_VERSION=${OPTARG};;
+        a) VLLM_VERSION=${OPTARG};;
+    esac
+done
 
+echo "Triton version is ${TRITON_CONTAINER_VERSION} and vllm version is ${VLLM_VERSION}"
 docker pull nvcr.io/nvidia/tritonserver:${TRITON_CONTAINER_VERSION}-vllm-python-py3