fix pretrain

Former-commit-id: a04115ec27
This commit is contained in:
hiyouga
2023-07-13 23:41:54 +08:00
parent 2b0fced03b
commit 32ec440d2e

View File

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