ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

recover logging

Browse Source 
Former-commit-id: d74014496e4ccda2de4482075a91747854facddd
This commit is contained in:
hiyouga
2023-06-06 21:36:37 +08:00
parent bf5ad34196
commit 12094c1db5
6 changed files with 17 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/utils/common.py
View File

@@ -38,7 +38,7 @@ from .config import (
)
from .other import (
get_main_logger,
get_logger,
load_trainable_params,
load_valuehead_params,
print_trainable_params,
@@ -53,7 +53,7 @@ require_version("peft>=0.3.0", "To fix: pip install peft>=0.3.0")
require_version("trl>=0.4.1", "To fix: pip install trl>=0.4.1")
logger = get_main_logger(__name__)
logger = get_logger(__name__)
def _init_adapter(
@@ -289,7 +289,7 @@ def prepare_args(
logger.info(
f"Process rank: {training_args.local_rank}, device: {training_args.device}, n_gpu: {training_args.n_gpu}\n"
+ f" distributed training: {bool(training_args.local_rank != -1)}, 16-bits training: {training_args.fp16}"
, main_process_only=False)
)
logger.info(f"Training/evaluation parameters {training_args}")
# Set seed before initializing model.