ci: Update release.yml (#11698)

* ci: Update release.yml

Signed-off-by: oliver könig <[email protected]>

* Update release.yml

Signed-off-by: oliver könig <[email protected]>

---------

Signed-off-by: oliver könig <[email protected]>

Author: ko3n1g

.github/workflows/release.yml

@@ -20,6 +20,10 @@ on:
         description: Ref (SHA or branch name) to release
         required: true
         type: string
+      version-bump-branch:
+        description: Branch for version bump
+        required: true
+        type: string
       dry-run:
         description: Do not publish a wheel and GitHub release.
         required: true
@@ -28,7 +32,7 @@ on:
 
 jobs: 
   release:
-    uses: NVIDIA/NeMo-FW-CI-templates/.github/workflows/[email protected].1
+    uses: NVIDIA/NeMo-FW-CI-templates/.github/workflows/[email protected].2
     with:
       release-ref: ${{ inputs.release-ref }}
       image-name: nemo_container
@@ -41,6 +45,7 @@ jobs:
       container-workdir: /workspace
       library-name: Neural Modules
       dry-run: ${{ inputs.dry-run }}
+      version-bump-branch: ${{ inputs.version-bump-branch }}
     secrets:
       TWINE_USERNAME: ${{ secrets.TWINE_USERNAME }}
       TWINE_PASSWORD: ${{ secrets.TWINE_PASSWORD }}