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

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/pairwise.py
View File

@@ -5,9 +5,9 @@ from .data_collator import DynamicDataCollatorWithPadding
from .peft_trainer import PeftTrainer
from .other import get_logger
from .other import get_main_logger
logger = get_logger(__name__)
logger = get_main_logger(__name__)
class PairwiseDataCollatorWithPadding(DynamicDataCollatorWithPadding):