Use proper attribute for sagemaker request (#11362)

2026-08-09 16:24:38 +00:00 · 2025-06-04 22:47:43 -07:00
parent 4019f79808
commit d7982bb0af
1 changed files with 1 additions and 1 deletions
@@ -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"""