diff --git a/.github/workflows/manual-freeze-tag.yml b/.github/workflows/manual-freeze-tag.yml
index 71e427482..90bdb3782 100644
--- a/.github/workflows/manual-freeze-tag.yml
+++ b/.github/workflows/manual-freeze-tag.yml
@@ -11,6 +11,11 @@ on:
         description: "Tag to apply to images"
         required: true
         type: string
+      text-generation-inference-tag:
+        default: "latest"
+        description: "Tag of ghcr.io/huggingface/text-generation-inference"
+        required: true
+        type: string
 
 jobs:
   freeze-tag:
@@ -36,6 +41,7 @@ jobs:
       - name: Run script
         run: |
           find . -name "*.md" | xargs sed -i "s|^docker\ compose|TAG=${{ github.event.inputs.tag }}\ docker\ compose|g"
+          find . -name "*.md" | xargs sed -i "s|ghcr.io/huggingface/text-generation-inference:latest[a-Z0-9\-]*|ghcr.io/huggingface/text-generation-inference:${{ github.event.inputs.text-generation-inference-tag }}|g"
 
       - name: Commit changes
         run: |
diff --git a/.github/workflows/scripts/freeze_images.sh b/.github/workflows/scripts/freeze_images.sh
index 0bdfa9fb2..7cd94ebd7 100644
--- a/.github/workflows/scripts/freeze_images.sh
+++ b/.github/workflows/scripts/freeze_images.sh
@@ -5,13 +5,13 @@
 
 declare -A dict
 dict["langchain/langchain"]="docker://docker.io/langchain/langchain"
-# dict["ghcr.io/huggingface/text-generation-inference"]="docker://ghcr.io/huggingface/text-generation-inference"
+dict["ghcr.io/huggingface/text-generation-inference"]="docker://ghcr.io/huggingface/text-generation-inference"
 
 function get_latest_version() {
     repo_image=$1
     versions=$(skopeo list-tags ${dict[$repo_image]} | jq -r '.Tags[]')
     printf "version list:\n$versions\n"
-    latest_version=$(printf "%s\n" "${versions[@]}" | grep -E '^[\.rc0-9\-]+$' | sort -V | tail -n 1)
+    latest_version=$(printf "%s\n" "${versions[@]}" | grep -E '^[\.0-9]+$' | sort -V | tail -n 1)
     echo "latest version: $latest_version"
     replace_image_version $repo_image $latest_version
 }
@@ -23,7 +23,8 @@ function replace_image_version() {
         echo "version is empty"
     else
         echo "replace $repo_image:latest with $repo_image:$version"
-        find . -name "Dockerfile" | xargs sed -i "s|$repo_image:latest|$repo_image:$version|g"
+        find . -name "Dockerfile" | xargs sed -i "s|$repo_image:latest.*|$repo_image:$version|g"
+        find . -name "*.yaml" | xargs sed -i "s|$repo_image:latest[a-Z0-9\-]*|$repo_image:$version|g"
     fi
 }