diff --git a/litellm/llms/sagemaker/completion/handler.py b/litellm/llms/sagemaker/completion/handler.py index ebd96ac5b1..3d4108776c 100644 --- a/litellm/llms/sagemaker/completion/handler.py +++ b/litellm/llms/sagemaker/completion/handler.py @@ -626,7 +626,7 @@ class SagemakerLLM(BaseAWSLLM): inference_params[k] = v #### HF EMBEDDING LOGIC - data = json.dumps({"text_inputs": input}).encode("utf-8") + data = json.dumps({"inputs": input}).encode("utf-8") ## LOGGING request_str = f"""