Use llmtuner logger
This commit is contained in:
parent
26817143ff
commit
fc70a92cb6
|
@ -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 (
|
||||
|
|
Loading…
Reference in New Issue