From 5c89fe2516a7df745340481e94c4dc363613c3a4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cl=C3=A9mentine=20Fourrier?= <22726840+clefourrier@users.noreply.github.com> Date: Wed, 18 Dec 2024 11:53:11 +0100 Subject: [PATCH] Update src/lighteval/models/transformers/transformers_model.py Co-authored-by: Albert Villanova del Moral <8515462+albertvillanova@users.noreply.github.com> --- src/lighteval/models/transformers/transformers_model.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/lighteval/models/transformers/transformers_model.py b/src/lighteval/models/transformers/transformers_model.py index e63d0840e..19ead82cc 100644 --- a/src/lighteval/models/transformers/transformers_model.py +++ b/src/lighteval/models/transformers/transformers_model.py @@ -240,8 +240,9 @@ class BaseModelConfig(TransformersModelConfig): def __post_init__(self): super().__post_init__() - logger.warning( - "Careful, BaseModelConfig is deprecated and will be removed, you should use TransformersModelConfig instead!" + warnings.warn( + "BaseModelConfig is deprecated and will be removed. Use TransformersModelConfig instead", + FutureWarning, )