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

fix pretrain

Browse Source 
Former-commit-id: a04115ec278e646347a6fec12f562494abd66e51
This commit is contained in:
hiyouga 2023-07-13 23:41:54 +08:00
parent 2b0fced03b
commit 32ec440d2e
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 import (
DynamicDataCollatorWithPadding,
PeftTrainer,
LogCallback,
load_pretrained,