From 5522a7eff12e137e17cc5befc948cdbb45028bce Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 21 Aug 2023 11:52:22 -0700 Subject: [PATCH] add replicate models in defaults --- litellm/__init__.py | 10 +++++++++- pyproject.toml | 2 +- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/litellm/__init__.py b/litellm/__init__.py index 498fb0eae8..6a71ff0af6 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -149,7 +149,15 @@ cohere_models = [ anthropic_models = ["claude-2", "claude-instant-1", "claude-instant-1.2"] replicate_models = [ - "replicate/" + "replicate/", + "replicate/llama-2-70b-chat:58d078176e02c219e11eb4da5a02a7830a283b14cf8f94537af893ccff5ee781", + "a16z-infra/llama-2-13b-chat:2a7f981751ec7fdf87b5b91ad4db53683a98082e9ff7bfd12c8cd5ea85980a52", + "joehoover/instructblip-vicuna13b:c4c54e3c8c97cd50c2d2fec9be3b6065563ccf7d43787fb99f84151b867178fe" + "replicate/dolly-v2-12b:ef0e1aefc61f8e096ebe4db6b2bacc297daf2ef6899f0f7e001ec445893500e5", + "a16z-infra/llama-2-7b-chat:7b0bfc9aff140d5b75bacbed23e91fd3c34b01a1e958d32132de6e0a19796e2c", + "replicate/vicuna-13b:6282abe6a492de4145d7bb601023762212f9ddbbe78278bd6771c8b3b2f2a13b", + "daanelson/flan-t5-large:ce962b3f6792a57074a601d3979db5839697add2e4e02696b3ced4c022d4767f", + "replit/replit-code-v1-3b:b84f4c074b807211cd75e3e8b1589b6399052125b4c27106e43d47189e8415ad" ] # placeholder, to make sure we accept any replicate model in our model_list openrouter_models = [ diff --git a/pyproject.toml b/pyproject.toml index 81bdd29c9b..fff93a92e5 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "litellm" -version = "0.1.441" +version = "0.1.442" description = "Library to easily interface with LLM API providers" authors = ["BerriAI"] license = "MIT License"