diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index 6de33a251..a833fad0d 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -5,6 +5,7 @@ on:
   push:
     branches:
       - 'main'
+      - 'update-gha'
     tags:
       - 'v*'
   pull_request:
@@ -34,6 +35,17 @@ jobs:
       - name: Set output
         id: vars
         run: echo "sha_short=$(git rev-parse --short HEAD)" >> $GITHUB_OUTPUT
+
+      - name: Docker meta
+        id: meta
+        uses: docker/metadata-action@v5
+        with:
+          images: |
+            ghcr.io/predibase/lorax
+          tags: |
+            type=semver,pattern={{version}}
+            type=semver,pattern={{major}}.{{minor}}
+            type=sha,prefix=,suffix=,format=short
       
       - name: Log in to GitHub Container Registry
         uses: docker/login-action@v1