423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-17 20:30:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

recover logging

Browse Source
This commit is contained in:
hiyouga
2023-06-06 21:36:37 +08:00
parent 4eb17bcf6c
commit 13d1f0709c
6 changed files with 17 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/utils/other.py
View File

@@ -10,8 +10,6 @@ from transformers.modeling_utils import PreTrainedModel
from transformers.generation.utils import LogitsProcessorList
from transformers.generation.logits_process import LogitsProcessor
from accelerate.logging import get_logger
from peft.utils.other import WEIGHTS_NAME
@@ -20,16 +18,19 @@ VALUE_HEAD_FILE_NAME = "value_head.bin"
FINETUNING_ARGS_NAME = "finetuning_args.json"
logger = get_logger(__name__, log_level="INFO")
def get_logger(name: str) -> logging.Logger:
return logging.getLogger(name)
logging.basicConfig(
format="%(asctime)s - %(levelname)s - %(name)s - %(message)s",
datefmt="%m/%d/%Y %H:%M:%S",
level=logging.INFO,
handlers=[logging.StreamHandler(sys.stdout)]
)
def get_main_logger(name: str) -> logging.Logger:
return get_logger(name, log_level="INFO")
logger = get_logger(__name__)
class AverageMeter: