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

fix pretrain

Browse Source 
Former-commit-id: 75b584875fc75c437491a5155ef310248e9b9dd4
This commit is contained in:
hiyouga 2023-07-13 23:41:54 +08:00
parent 25182c4779
commit fd956c218b
1 changed files with 0 additions and 1 deletions
Show Stats 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,