Update src/lighteval/models/transformers/transformers_model.py

Co-authored-by: Nathan Habib <[email protected]>
huggingface · Dec 16, 2024 · be99c5e · be99c5e
1 parent fb4ecdc
commit be99c5e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()