From 7d1fe50977e76640ce494c0db85bd908ac1eec3e Mon Sep 17 00:00:00 2001 From: yhyu13 Date: Sat, 16 Dec 2023 07:15:27 +0000 Subject: [PATCH] Use llmtuner logger Former-commit-id: ef5a560b4246e04e0ef2612e3520e05288e93707 --- src/llmtuner/extras/misc.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/llmtuner/extras/misc.py b/src/llmtuner/extras/misc.py index 4f123e14..9a50e369 100644 --- a/src/llmtuner/extras/misc.py +++ b/src/llmtuner/extras/misc.py @@ -5,8 +5,8 @@ import torch from typing import TYPE_CHECKING, Any, Dict, Optional, Tuple from transformers import InfNanRemoveLogitsProcessor, LogitsProcessorList -import logging -logger = logging.getLogger(__name__) +from llmtuner.extras.logging import get_logger +logger = get_logger(__name__) try: from transformers.utils import (