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:
parent
c6e86d5e77
commit
5777ccd03e
1 changed files with 16 additions and 0 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue