phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add flan + vicuna + fix replicate errors

Browse source
This commit is contained in:
ishaan-jaff 2023-09-06 11:21:57 -07:00
parent 48ee4a08ac
commit 1da6026622
3 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/replicate.py
View file

@ -39,7 +39,7 @@ def start_prediction(version_id, input_data, api_token, logging_obj):
response_data = response.json()
return response_data.get("urls", {}).get("get")
else:
raise ReplicateError(response.status_code, message=response.text)
raise ReplicateError(response.status_code, f"Failed to start prediction {response.text}")
# Function to handle prediction response (non-streaming)
def handle_prediction_response(prediction_url, api_token, print_verbose):