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

support distributed quantized training

Browse Source
This commit is contained in:
hiyouga
2023-06-06 17:39:41 +08:00
parent 3d8d5ee5d5
commit 4eb17bcf6c
7 changed files with 20 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/utils/peft_trainer.py
View File

@@ -21,7 +21,7 @@ from peft.utils.other import WEIGHTS_NAME
from .config import FinetuningArguments
from .other import (
get_logger,
get_main_logger,
get_state_dict,
load_trainable_params,
load_valuehead_params,
@@ -30,7 +30,7 @@ from .other import (
)
logger = get_logger(__name__)
logger = get_main_logger(__name__)
class LogCallback(TrainerCallback):