Merge branch 'main' into add_litellm_inference

JoelNiklaus · Dec 17, 2024 · 3ebff6c · 3ebff6c
2 parents 5115403 + bc62cb9
commit 3ebff6c
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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