kyy/llm_trainer
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix pretrain

Browse Source
This commit is contained in:
hiyouga committed 2023-07-13 23:41:54 +08:00
1 parent 08439d29b2
commit a04115ec27
1 file changed
-1
Download Patch File Download Diff File Expand all files Collapse all files
src/train_pt.py
-1
View File
@@ -9,7 +9,6 @@ from transformers import DataCollatorForSeq2Seq
from utils.other import IGNORE_INDEX
from utils import (
DynamicDataCollatorWithPadding,
PeftTrainer,
LogCallback,
load_pretrained,