diff --git a/tests/models/test_endpoint_model.py b/tests/models/test_endpoint_model.py index 29fbb3c48..f4ba15d91 100644 --- a/tests/models/test_endpoint_model.py +++ b/tests/models/test_endpoint_model.py @@ -53,7 +53,7 @@ class TestInferenceEndpointModelConfig: }, ), ( - "examples/model_configs/endpoint_model_lite.yaml", + "examples/model_configs/serverless_model.yaml", { "model_name": "meta-llama/Llama-3.1-8B-Instruct", # Defaults: