Formatting

Author: amzn-choeric

src/sagemaker/huggingface/llm_utils.py

@@ -57,15 +57,15 @@ def get_huggingface_llm_image_uri(
             version=version,
             image_scope="inference",
         )
-    elif backend == "huggingface-neuronx":
+    if backend == "huggingface-neuronx":
         return image_uris.retrieve(
             "huggingface-llm-neuronx",
             region=region,
             version=version,
             image_scope="inference",
             inference_tool="neuronx",
         )
-    elif backend == "lmi":
+    if backend == "lmi":
         version = version or "0.24.0"
         return image_uris.retrieve(framework="djl-deepspeed", region=region, version=version)
     raise ValueError("Unsupported backend: %s" % backend)