Migrate to modern Python Logger API

Signed-off-by: Emmanuel Ferdman <emmanuelferdman@gmail.com>
This commit is contained in:
Emmanuel Ferdman
2025-05-12 14:09:26 -07:00
parent ef2a7abad4
commit a3a35acc7e
2 changed files with 2 additions and 2 deletions

View File

@@ -413,7 +413,7 @@ class BertEncoder(nn.Module):
if self.gradient_checkpointing and self.training:
if use_cache:
logger.warn(
logger.warning(
"`use_cache=True` is incompatible with gradient checkpointing. Setting `use_cache=False`..."
)
use_cache = False