diff --git a/.github/workflows/docker-build.yaml b/.github/workflows/docker-build.yaml
index b3fa778a..7f043c24 100644
--- a/.github/workflows/docker-build.yaml
+++ b/.github/workflows/docker-build.yaml
@@ -66,6 +66,7 @@ jobs:
 
       - name: Build and push Docker image with CUDA
         uses: docker/build-push-action@v5
+        if: github.ref == 'refs/heads/main'
         with:
           context: .
           push: true
@@ -75,6 +76,7 @@ jobs:
 
       - name: Build and push Docker image with Ollama
         uses: docker/build-push-action@v5
+        if: github.ref == 'refs/heads/main'
         with:
           context: .
           push: true