diff --git a/tests/local_testing/test_completion_cost.py b/tests/local_testing/test_completion_cost.py index d82909e135..c15922e212 100644 --- a/tests/local_testing/test_completion_cost.py +++ b/tests/local_testing/test_completion_cost.py @@ -1266,7 +1266,7 @@ def test_completion_cost_prompt_caching(model, custom_llm_provider): @pytest.mark.parametrize( "model", [ - "databricks/databricks-meta-llama-3.3-70b-instruct", + "databricks/databricks-meta-llama-3.2-3b-instruct", "databricks/databricks-meta-llama-3-70b-instruct", "databricks/databricks-dbrx-instruct", # "databricks/databricks-mixtral-8x7b-instruct",