Add TEI Serving

Jonathan Makunga · Jonathan Makunga · commit 7ee2e44839bd · 2024-05-17T14:05:34.000-07:00
diff --git a/src/sagemaker/serve/builder/tei_builder.py b/src/sagemaker/serve/builder/tei_builder.py
@@ -216,7 +216,7 @@ def _build_for_tei(self):
         """Placeholder docstring"""
         self.secret_key = None
 
-        self._set_to_tgi()
+        self._set_to_tei()
 
         self.pysdk_model = self._build_for_hf_tei()
         return self.pysdk_model
