From be99c5ed89c5d1a26ebf6d27f9d2262d4aab3149 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cl=C3=A9mentine=20Fourrier?= <22726840+clefourrier@users.noreply.github.com> Date: Mon, 16 Dec 2024 14:43:20 +0100 Subject: [PATCH] Update src/lighteval/models/transformers/transformers_model.py Co-authored-by: Nathan Habib <30601243+NathanHB@users.noreply.github.com> --- src/lighteval/models/transformers/transformers_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lighteval/models/transformers/transformers_model.py b/src/lighteval/models/transformers/transformers_model.py index 69af4733..e63d0840 100644 --- a/src/lighteval/models/transformers/transformers_model.py +++ b/src/lighteval/models/transformers/transformers_model.py @@ -277,7 +277,7 @@ def __init__( logger.info("Compiling the model") self.model.model.compile() except AttributeError as e: - logger.warn("Could not compile the model because: ", e) + logger.warning("Could not compile the model because: ", e) self.model_name = _simplify_name(config.pretrained) self.model_sha = config.get_model_sha()