diff --git a/.gitea/workflows/build.yml b/.gitea/workflows/build.yml
index 09af056..9536097 100644
--- a/.gitea/workflows/build.yml
+++ b/.gitea/workflows/build.yml
@@ -3,6 +3,7 @@ name: Build ROCm Image
 on:
   push:
     branches:
+      - main
       - onxx
 
 jobs:
@@ -22,14 +23,21 @@ jobs:
           username: ${{ secrets.REGISTRY_USERNAME }}
           password: ${{ secrets.REGISTRY_TOKEN }}
 
+      - name: Set tags
+        id: tags
+        run: |
+          if [ "${{ gitea.ref_name }}" = "main" ]; then
+            echo "tags=git.sdgarren.com/scott/rocm-chatterbox-whisper:latest,git.sdgarren.com/scott/rocm-chatterbox-whisper:${{ gitea.sha }}" >> $GITHUB_OUTPUT
+          else
+            echo "tags=git.sdgarren.com/scott/rocm-chatterbox-whisper:${{ gitea.ref_name }},git.sdgarren.com/scott/rocm-chatterbox-whisper:${{ gitea.sha }}" >> $GITHUB_OUTPUT
+          fi
+
       - name: Build and Push
         uses: docker/build-push-action@v6
         with:
           context: .
           file: Dockerfile.rocm
           push: true
-          tags: |
-            git.sdgarren.com/scott/rocm-chatterbox-whisper:latest
-            git.sdgarren.com/scott/rocm-chatterbox-whisper:${{ gitea.sha }}
+          tags: ${{ steps.tags.outputs.tags }}
           cache-from: type=registry,ref=git.sdgarren.com/scott/rocm-chatterbox-whisper:latest
           cache-to: type=inline