storyteller-ml/animation/hallo-83dd4eb/docker-56038fd/docker-bake.hcl

variable "REGISTRY" {
    default = "docker.io"
}

variable "REGISTRY_USER" {
    default = "ashleykza"
}
variable "APP" {
    default = "hallo"
}

variable "RELEASE" {
    default = "1.2.0"
}

variable "CU_VERSION" {
    default = "121"
}


variable "BASE_IMAGE" {
    default = "12.2.2-cudnn8-devel-ubuntu22.04"
}

variable "CUDA_VERSION" {
    default = "12.1.1"
}

variable "TORCH_VERSION" {
    default = "2.2.2"
}

target "default" {
    dockerfile = "Dockerfile"
    tags = ["${REGISTRY}/${REGISTRY_USER}/${APP}:${RELEASE}"]
    args = {
        RELEASE = "${RELEASE}"
        BASE_IMAGE = "${BASE_IMAGE_REPOSITORY}:${BASE_IMAGE_VERSION}-cuda${CUDA_VERSION}-torch${TORCH_VERSION}"
        INDEX_URL = "https://download.pytorch.org/whl/cu${CU_VERSION}"
        TORCH_VERSION = "${TORCH_VERSION}+cu${CU_VERSION}"
        XFORMERS_VERSION = "0.0.25.post1"
        HALLO_VERSION = "v1.0.0"
    }
}