Merge branch 'main' into feature/damian/kv_cache_bloom

neuralmagic · Jul 12, 2023 · 9c5e27d · 9c5e27d
2 parents 6ad363c + 6e761ed
commit 9c5e27d
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/.github/workflows/build-and-publish-release-images.yaml b/.github/workflows/build-and-publish-release-images.yaml
@@ -43,7 +43,19 @@ jobs:
         run: |
           echo ${{ steps.extract_tag.outputs.tag }}
 
-      - name: Build and  push sparseml with all dependencies and default cuda 11.1.1
+      - name: Build and push sparseml latest using default cuda 11.1.1
+        if: ${{ startsWith(github.ref, 'refs/tags/v') }}
+        uses: docker/build-push-action@v2
+        with:
+          context: ./docker
+          build-args: |
+            DEPS=all
+            VERSION=${{ steps.extract_tag.outputs.tag }}
+          push: true
+          tags: |
+            ghcr.io/neuralmagic/sparseml:latest
+
+      - name: Build and push sparseml with all dependencies and default cuda 11.1.1
         if: ${{ startsWith(github.ref, 'refs/tags/v') }}
         uses: docker/build-push-action@v2
         with:
@@ -55,7 +67,7 @@ jobs:
           tags: |
             ghcr.io/neuralmagic/sparseml:${{ steps.extract_tag.outputs.tag }}
 
-      - name: Build and  push sparseml with all dependencies and cuda 10.2
+      - name: Build and push sparseml with all dependencies and cuda 10.2
         if: ${{ startsWith(github.ref, 'refs/tags/v') }}
         uses: docker/build-push-action@v2
         with: