Update build_docker.sh

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