diff --git a/src/lighteval/models/endpoints/endpoint_model.py b/src/lighteval/models/endpoints/endpoint_model.py index 18fd7d772..80798b616 100644 --- a/src/lighteval/models/endpoints/endpoint_model.py +++ b/src/lighteval/models/endpoints/endpoint_model.py @@ -224,9 +224,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