fix typo

haixiw · haixiw · commit 92af5692bd63 · 2024-05-20T19:14:05.000Z
diff --git a/src/sagemaker/huggingface/llm_utils.py b/src/sagemaker/huggingface/llm_utils.py
@@ -74,7 +74,7 @@ def get_huggingface_llm_image_uri(
         )
     if backend == "huggingface-tei-cpu":
         return image_uris.retrieve(
-            "huggingface-tei",
+            "huggingface-tei-cpu",
             region=region,
             version=version,
             image_scope="inference",

Original file line number	Diff line number	Diff line change
`@@ -74,7 +74,7 @@ def get_huggingface_llm_image_uri(`
`74`	`74`	`)`
`75`	`75`	`if backend == "huggingface-tei-cpu":`
`76`	`76`	`return image_uris.retrieve(`
`77`		`- "huggingface-tei",`
	`77`	`+ "huggingface-tei-cpu",`
`78`	`78`	`region=region,`
`79`	`79`	`version=version,`
`80`	`80`	`image_scope="inference",`