Use llmtuner logger

Former-commit-id: fc70a92cb6e9c22bab9a0695f476ae80461c656f
This commit is contained in:
yhyu13 2023-12-16 07:15:27 +00:00
parent 362e3c913f
commit cc91724507

View File

@ -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 (