From c52159afad14f84bfc2e34aaec2f76a0fef994de Mon Sep 17 00:00:00 2001
From: sirutBuasai <sirutbuasai27@outlook.com>
Date: Fri, 10 Oct 2025 15:19:31 -0700
Subject: [PATCH] fix TEI config

---
 src/sagemaker/image_uri_config/huggingface-tei-cpu.json | 3 ++-
 src/sagemaker/image_uri_config/huggingface-tei.json     | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/sagemaker/image_uri_config/huggingface-tei-cpu.json b/src/sagemaker/image_uri_config/huggingface-tei-cpu.json
index 3bd960c45d..d693106c1d 100644
--- a/src/sagemaker/image_uri_config/huggingface-tei-cpu.json
+++ b/src/sagemaker/image_uri_config/huggingface-tei-cpu.json
@@ -7,7 +7,8 @@
             "1.2": "1.2.3",
             "1.4": "1.4.0",
             "1.6": "1.6.0",
-            "1.7": "1.7.0"
+            "1.7": "1.7.0",
+            "1.8": "1.8.2"
         },
         "versions": {
             "1.2.3": {
diff --git a/src/sagemaker/image_uri_config/huggingface-tei.json b/src/sagemaker/image_uri_config/huggingface-tei.json
index 9b4e4773d4..2df6abea11 100644
--- a/src/sagemaker/image_uri_config/huggingface-tei.json
+++ b/src/sagemaker/image_uri_config/huggingface-tei.json
@@ -7,7 +7,8 @@
             "1.2": "1.2.3",
             "1.4": "1.4.0",
             "1.6": "1.6.0",
-            "1.7": "1.7.0"
+            "1.7": "1.7.0",
+            "1.8": "1.8.2"
         },
         "versions": {
             "1.2.3": {