66 lines
1.3 KiB
HCL
66 lines
1.3 KiB
HCL
variable "PUSH" {
|
|
default = "true"
|
|
}
|
|
|
|
variable "REPOSITORY" {
|
|
default = "runpod"
|
|
}
|
|
|
|
variable "BASE_IMAGE_VERSION" {
|
|
default = "1.1.0preview"
|
|
}
|
|
|
|
group "all" {
|
|
targets = ["base", "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}"]
|
|
}
|
|
|
|
target "worker-1210" {
|
|
tags = ["${REPOSITORY}/worker-vllm:${BASE_IMAGE_VERSION}-cuda12.1.0"]
|
|
context = "."
|
|
dockerfile = "Dockerfile"
|
|
args = {
|
|
BASE_IMAGE_VERSION = "${BASE_IMAGE_VERSION}"
|
|
WORKER_CUDA_VERSION = "12.1.0"
|
|
}
|
|
output = ["type=docker,push=${PUSH}"]
|
|
}
|