Merge branch 'main' into fix422

huggingface · Dec 17, 2024 · b71f84f · b71f84f
2 parents d437f03 + bc62cb9
commit b71f84f
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
@@ -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