From 14cacd55fe8094b62e3d39afcd7e936cc5fb393d Mon Sep 17 00:00:00 2001 From: pandyamarut Date: Wed, 31 Jul 2024 18:44:46 -0700 Subject: [PATCH] update docker Signed-off-by: pandyamarut --- docker-bake.hcl | 43 +++++-------------------------------------- 1 file changed, 5 insertions(+), 38 deletions(-) diff --git a/docker-bake.hcl b/docker-bake.hcl index 2830d6f..b121032 100644 --- a/docker-bake.hcl +++ b/docker-bake.hcl @@ -7,54 +7,21 @@ variable "REPOSITORY" { } variable "BASE_IMAGE_VERSION" { - default = "1.1.0preview" + default = "v1.1preview" } group "all" { - targets = ["base", "main"] + targets = ["main"] } -group "base" { - targets = ["base-1180", "base-1210"] -} group "main" { - targets = ["worker-1180", "worker-1210"] -} - -target "base-1180" { - tags = ["${REPOSITORY}/worker-vllm:base-${BASE_IMAGE_VERSION}-cuda11.8.0"] - context = "vllm-base-image" - dockerfile = "Dockerfile" - args = { - WORKER_CUDA_VERSION = "11.8.0" - } - output = ["type=docker,push=${PUSH}"] -} - -target "base-1210" { - tags = ["${REPOSITORY}/worker-vllm:base-${BASE_IMAGE_VERSION}-cuda12.1.0"] - context = "vllm-base-image" - dockerfile = "Dockerfile" - args = { - WORKER_CUDA_VERSION = "12.1.0" - } - output = ["type=docker,push=${PUSH}"] -} - -target "worker-1180" { - tags = ["${REPOSITORY}/worker-vllm:${BASE_IMAGE_VERSION}-cuda11.8.0"] - context = "." - dockerfile = "Dockerfile" - args = { - BASE_IMAGE_VERSION = "${BASE_IMAGE_VERSION}" - WORKER_CUDA_VERSION = "11.8.0" - } - output = ["type=docker,push=${PUSH}"] + targets = ["worker-1210"] } + target "worker-1210" { - tags = ["${REPOSITORY}/worker-vllm:${BASE_IMAGE_VERSION}-cuda12.1.0"] + tags = ["${REPOSITORY}/worker-v1-vllm:${BASE_IMAGE_VERSION}-cuda12.1.0"] context = "." dockerfile = "Dockerfile" args = {