diff --git a/src/lighteval/models/endpoints/endpoint_model.py b/src/lighteval/models/endpoints/endpoint_model.py index 1344e248..e50c0405 100644 --- a/src/lighteval/models/endpoints/endpoint_model.py +++ b/src/lighteval/models/endpoints/endpoint_model.py @@ -216,9 +216,7 @@ def __init__( # noqa: C901 **config.get_dtype_args(), **config.get_custom_env_vars(), }, - "url": ( - config.image_url or "ghcr.io/huggingface/text-generation-inference:latest" - ), + "url": (config.image_url or "ghcr.io/huggingface/text-generation-inference:3.0.1"), }, ) else: # Endpoint exists