ci: Update release.yaml (#471)

NVIDIA · Jan 3, 2025 · d0e1fa4 · d0e1fa4
1 parent 35fcfd9
commit d0e1fa4
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/.github/workflows/release.yaml b/.github/workflows/release.yaml
@@ -25,10 +25,14 @@ on:
         required: true
         default: true
         type: boolean
+      version-bump-branch:
+        description: Branch for version bump
+        required: true
+        type: string
 
 jobs:
   release:
-    uses: NVIDIA/NeMo-FW-CI-templates/.github/workflows/[email protected].1
+    uses: NVIDIA/NeMo-FW-CI-templates/.github/workflows/[email protected].3
     with:
       release-ref: ${{ inputs.release-ref }}
       image-name: nemo_aligner_container
@@ -42,6 +46,7 @@ jobs:
       container-workdir: /opt/NeMo-Aligner
       library-name: NeMo-Aligner
       dry-run: ${{ inputs.dry-run }}
+      version-bump-branch: ${{ inputs.version-bump-branch }}
     secrets:
       TWINE_USERNAME: ${{ secrets.TWINE_USERNAME }}
       TWINE_PASSWORD: ${{ secrets.TWINE_PASSWORD }}