From 68a35ffce31f246c7cdbb5b6683a403cce8ab9a9 Mon Sep 17 00:00:00 2001 From: Daniel King Date: Fri, 15 Dec 2023 14:12:15 -0800 Subject: [PATCH] remove handler --- scripts/train/train.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/scripts/train/train.py b/scripts/train/train.py index b801ad8f28..2c1099ff00 100644 --- a/scripts/train/train.py +++ b/scripts/train/train.py @@ -21,7 +21,6 @@ from omegaconf import DictConfig, ListConfig from omegaconf import OmegaConf as om from transformers import PreTrainedTokenizerBase -from transformers.utils import logging as transformers_logging from llmfoundry import (COMPOSER_MODEL_REGISTRY, ComposerHFCausalLM, MPTForCausalLM) @@ -37,9 +36,6 @@ log = logging.getLogger(__name__) -transformers_logging.enable_default_handler() -transformers_logging.enable_explicit_format() - def validate_config(cfg: DictConfig): """Validates compatible model and dataloader selection."""