Skip to content

Commit

Permalink
remove handler
Browse files Browse the repository at this point in the history
  • Loading branch information
dakinggg committed Dec 15, 2023
1 parent 3248305 commit 68a35ff
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions scripts/train/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand All @@ -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."""
Expand Down

0 comments on commit 68a35ff

Please sign in to comment.