mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-03 20:22:49 +08:00
Use llmtuner logger
Former-commit-id: fc70a92cb6e9c22bab9a0695f476ae80461c656f
This commit is contained in:
parent
362e3c913f
commit
cc91724507
@ -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…
x
Reference in New Issue
Block a user