From 4663ba3be0de64dc5012f7d2f99edb5045ff8bbe Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 26 Jun 2025 03:39:13 -0700 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index e03f50baa9..34a61e250d 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -153,6 +153,8 @@ from transformers.training_args import logger as transformers_training_args_logg transformers_training_args_logger.addFilter(HideLoggingMessage("The speedups")) # torch.distributed process group is initialized, but parallel_mode != ParallelMode.DISTRIBUTED. transformers_training_args_logger.addFilter(HideLoggingMessage("torch.distributed")) +# average_tokens_across_devices is set to True but it is invalid when world size is1 +transformers_training_args_logger.addFilter(HideLoggingMessage("average_tokens_across_devices")) del transformers_training_args_logger # No label_names provided for model class