33 lines
514 B
HCL
33 lines
514 B
HCL
variable "PUSH" {
|
|
default = "true"
|
|
}
|
|
|
|
variable "REPOSITORY" {
|
|
default = "runpod"
|
|
}
|
|
|
|
variable "BASE_IMAGE_VERSION" {
|
|
default = "stable"
|
|
}
|
|
|
|
group "all" {
|
|
targets = ["main"]
|
|
}
|
|
|
|
|
|
group "main" {
|
|
targets = ["worker-1210"]
|
|
}
|
|
|
|
|
|
target "worker-1210" {
|
|
tags = ["${REPOSITORY}/worker-v1-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}"]
|
|
}
|