Use llmtuner logger
Former-commit-id: ef5a560b4246e04e0ef2612e3520e05288e93707
This commit is contained in:
@@ -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 (
|
||||
|
||||
Reference in New Issue
Block a user