branch name

mosaicml · Oct 19, 2024 · 742541e · 742541e
1 parent aebce3b
commit 742541e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/release.yaml b/.github/workflows/release.yaml
@@ -73,7 +73,7 @@ jobs:
         cache-from: type=registry,ref=${{ env.IMAGE_CACHE }},scope=global
         cache-to: type=registry,ref=${{ env.IMAGE_CACHE }},mode=max,scope=global
         build-args: |
-          BRANCH_NAME=main
+          BRANCH_NAME=${{ github.ref_name }}
           BASE_IMAGE=${{ matrix.base_image }}
           DEP_GROUPS=${{ matrix.dep_groups }}
           TE_COMMIT=${{ matrix.te_commit }}