added more tags

vectorch-ai · May 14, 2024 · 2209517 · 2209517
1 parent 948128d
commit 2209517
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/publish_base_image.yml b/.github/workflows/publish_base_image.yml
@@ -32,6 +32,7 @@ jobs:
             CUDA_VERSION=12.1
           tags: |
             vectorchai/scalellm_builder:cuda12.1-ubuntu22.04
+            vectorchai/scalellm_builder:cuda12.1
 
       - name: Build base for cuda 11.8
         uses: docker/build-push-action@v5
@@ -44,4 +45,5 @@ jobs:
             CUDA_VERSION=11.8
           tags: |
             vectorchai/scalellm_builder:cuda11.8-ubuntu22.04
+            vectorchai/scalellm_builder:cuda11.8