Use proper attribute for sagemaker request (#11362)

This commit is contained in:
Tom Bocklisch 2025-06-05 07:47:43 +02:00 committed by GitHub
parent 4019f79808
commit d7982bb0af
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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"""