echo vars

.github/workflows/tpu-tgi-release.yml

@@ -6,6 +6,7 @@ on:
   push:
     branches: tgi-docker-workflow
 
+
 jobs:
   docker:
     name: Push TGI Docker container to Docker Hub and Github Registry
@@ -43,6 +44,12 @@ jobs:
               prefix=
               suffix=-tgi
 
+        - name: Get the version
+          id: version
+          run: |
+           VERSION=$(awk '/__version__ = "(.*)"/{print $3}' optimum/tpu/version.py | sed 's/"//g')
+           echo "version=$VERSION" >> $GITHUB_OUTPUT
+
         - name: Build and push Docker image
           id: push
           uses: docker/build-push-action@v6
@@ -53,7 +60,7 @@ jobs:
             tags: ${{ steps.meta.outputs.tags }}
             labels: ${{ steps.meta.outputs.labels }}
             build-args: |
-              VERSION=${{ github.event.release.tag_name }}
+              VERSION=${{ steps.version.outputs.version }}
               TGI_VERSION=5bc3d65dd32ba1f979540caeccbf3dd8798dd9df
             target: inference-endpoint