diff --git a/.github/workflows/docker-publish.yml b/.github/workflows/docker-publish.yml
index c917965..0975b63 100644
--- a/.github/workflows/docker-publish.yml
+++ b/.github/workflows/docker-publish.yml
@@ -67,8 +67,7 @@ jobs:
           DOCKER_BUILDKIT: 1
           BUILDKIT_STEP_LOG_MAX_SIZE: 10485760
           VERSION: ${{ needs.prepare-release.outputs.version }}
-        run: |
-          docker buildx bake --push
+        run: docker buildx bake --push
 
   create-release:
     needs: [prepare-release, build-images]
diff --git a/docker-bake.hcl b/docker-bake.hcl
index bed8065..f030c18 100644
--- a/docker-bake.hcl
+++ b/docker-bake.hcl
@@ -15,11 +15,16 @@ variable "REPO" {
     default = "kokoro-fastapi"
 }
 
+variable "DOWNLOAD_MODEL" {
+    default = "true"
+}
+
 # Common settings shared between targets
 target "_common" {
     context = "."
     args = {
         DEBIAN_FRONTEND = "noninteractive"
+        DOWNLOAD_MODEL = "${DOWNLOAD_MODEL}"
     }
     cache-from = ["type=registry,ref=${REGISTRY}/${OWNER}/${REPO}-cache"]
     cache-to = ["type=registry,ref=${REGISTRY}/${OWNER}/${REPO}-cache,mode=max"]