mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-24 18:24:20 +00:00
databricks/databricks-meta-llama-3-3-70b-instruct
This commit is contained in:
parent
8d338aee78
commit
d8f47fc9e5
1 changed files with 2 additions and 2 deletions
|
@ -1266,13 +1266,13 @@ def test_completion_cost_prompt_caching(model, custom_llm_provider):
|
|||
@pytest.mark.parametrize(
|
||||
"model",
|
||||
[
|
||||
"databricks/databricks-meta-llama-3.2-3b-instruct",
|
||||
"databricks/databricks-meta-llama-3.3-70b-instruct",
|
||||
"databricks/databricks-meta-llama-3-3-70b-instruct",
|
||||
"databricks/databricks-dbrx-instruct",
|
||||
# "databricks/databricks-mixtral-8x7b-instruct",
|
||||
],
|
||||
)
|
||||
def test_completion_cost_databricks(model):
|
||||
litellm._turn_on_debug()
|
||||
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
|
||||
litellm.model_cost = litellm.get_model_cost_map(url="")
|
||||
model, messages = model, [{"role": "user", "content": "What is 2+2?"}]
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue