From bc62cb98e45a97a50a7cc76d5e60b184d3fb6ae1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Erik=20Kaunism=C3=A4ki?= Date: Tue, 17 Dec 2024 13:54:51 +0100 Subject: [PATCH] use concrete version instead of latest (#452) * recommended to use concrete version instead of latest * ruff style --------- Co-authored-by: Nathan Habib <30601243+NathanHB@users.noreply.github.com> --- src/lighteval/models/endpoints/endpoint_model.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/lighteval/models/endpoints/endpoint_model.py b/src/lighteval/models/endpoints/endpoint_model.py index 1344e2485..e50c0405c 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