diff --git a/docker-bake.hcl b/docker-bake.hcl new file mode 100644 index 0000000..d8184c2 --- /dev/null +++ b/docker-bake.hcl @@ -0,0 +1,32 @@ +variable "PUSH" { + default = "true" +} + +variable "REPOSITORY" { + default = "runpod" +} + +variable "BASE_IMAGE_VERSION" { + default = "preview" +} + +group "all" { + targets = ["main"] +} + + +group "main" { + targets = ["worker-1210"] +} + + +target "worker-1210" { + tags = ["${REPOSITORY}/worker-sglang:${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}"] +} \ No newline at end of file