From e536429f42a7c136bf823207c63fa8f0f324b50b Mon Sep 17 00:00:00 2001
From: Somasundaram <somasundaram.sindhu@gmail.com>
Date: Mon, 2 Dec 2024 16:07:56 -0800
Subject: [PATCH] change OnnxRuntime engine to Rust for text embedding

---
 src/sagemaker/djl_inference/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/sagemaker/djl_inference/model.py b/src/sagemaker/djl_inference/model.py
index 8c724a6502..2a23cba9ec 100644
--- a/src/sagemaker/djl_inference/model.py
+++ b/src/sagemaker/djl_inference/model.py
@@ -148,7 +148,7 @@ def _infer_engine(self) -> Optional[str]:
             return self.engine
 
         if self.task == "text-embedding":
-            return "OnnxRuntime"
+            return "Rust"
         return "Python"
 
     def _infer_image_uri(self):