From cc917245079f6c3c97df9fd4a03ce59033dc1fe2 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: fc70a92cb6e9c22bab9a0695f476ae80461c656f --- 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 (