+5
-38
@@ -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 = {
|
||||
|
||||
Reference in New Issue
Block a user