From 32ec440d2eb7f5a07dde603024e354470a3040fc Mon Sep 17 00:00:00 2001 From: hiyouga Date: Thu, 13 Jul 2023 23:41:54 +0800 Subject: [PATCH] fix pretrain Former-commit-id: a04115ec278e646347a6fec12f562494abd66e51 --- src/train_pt.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/train_pt.py b/src/train_pt.py index f7123684..7461e251 100644 --- a/src/train_pt.py +++ b/src/train_pt.py @@ -9,7 +9,6 @@ from transformers import DataCollatorForSeq2Seq from utils.other import IGNORE_INDEX from utils import ( - DynamicDataCollatorWithPadding, PeftTrainer, LogCallback, load_pretrained,