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

Co-authored-by: Albert Villanova del Moral <[email protected]>
huggingface · Dec 18, 2024 · 5c89fe2 · 5c89fe2
1 parent 8e3b7e2
commit 5c89fe2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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,
         )