From 5777ccd03e5875234c7eb57f12dfeabeec5d1569 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 3 Jun 2026 06:05:53 -0700 Subject: [PATCH] Logging cleanup (#5973) * Logging cleanup * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> --- unsloth/models/_utils.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index abb6969a7e..138a7b3cc0 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -782,6 +782,22 @@ transformers_trainer_logger.addFilter(HideLoggingMessage("No label_names")) transformers_trainer_logger.addFilter(HideLoggingMessage("The tokenizer has new")) del transformers_trainer_logger +# Strip the "[transformers] " prefix transformers>=5 adds; keep the messages. +try: + try: + import transformers.utils.logging as _tf_log + + _tf_log._configure_library_root_logger() + except Exception: + pass + _tf_root = logging.getLogger("transformers") + _tf_fmt = logging.Formatter("%(message)s") + for _h in list(_tf_root.handlers): + _h.setFormatter(_tf_fmt) + del _tf_root, _tf_fmt +except Exception: + pass + # Using the default loss: `ForCausalLMLoss`. try: from transformers.modeling_utils import logger as transformers_modeling_utils_logger