423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-14 19:06:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix pretrain

Browse Source
This commit is contained in:
hiyouga
2023-07-13 23:41:54 +08:00
parent 08439d29b2
commit a04115ec27
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/train_pt.py
View File

@@ -9,7 +9,6 @@ from transformers import DataCollatorForSeq2Seq
from utils.other import IGNORE_INDEX from utils.other import IGNORE_INDEX
from utils import ( from utils import (
DynamicDataCollatorWithPadding,
PeftTrainer, PeftTrainer,
LogCallback, LogCallback,
load_pretrained, load_pretrained,