423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-16 00:28:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use llmtuner logger

Browse Source 
Former-commit-id: ef5a560b4246e04e0ef2612e3520e05288e93707
This commit is contained in:
yhyu13 2023-12-16 07:15:27 +00:00
parent c0e5e3c5d5
commit 7d1fe50977
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/llmtuner/extras/misc.py
View File

@ -5,8 +5,8 @@ import torch
from typing import TYPE_CHECKING, Any, Dict, Optional, Tuple from typing import TYPE_CHECKING, Any, Dict, Optional, Tuple
from transformers import InfNanRemoveLogitsProcessor, LogitsProcessorList from transformers import InfNanRemoveLogitsProcessor, LogitsProcessorList
import logging from llmtuner.extras.logging import get_logger
logger = logging.getLogger(__name__) logger = get_logger(__name__)
try: try:
from transformers.utils import ( from transformers.utils import (