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>
This commit is contained in:
Daniel Han 2026-06-03 06:05:53 -07:00 committed by GitHub
commit 5777ccd03e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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