ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

(feat) replicate/deployments: add POST Req view

Browse Source
This commit is contained in:
ishaan-jaff 2023-12-04 13:43:03 -08:00
parent 07a2035651
commit 32ecc1a677
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
litellm/llms/replicate.py
View File

@ -100,7 +100,7 @@ def start_prediction(version_id, input_data, api_token, api_base, logging_obj, p
logging_obj.pre_call(
input=input_data["prompt"],
api_key="",
additional_args={"complete_input_dict": initial_prediction_data, "headers": headers},
additional_args={"complete_input_dict": initial_prediction_data, "headers": headers, "api_base": base_url},
)
response = requests.post(f"{base_url}/predictions", json=initial_prediction_data, headers=headers)