From 961e6a9ba487a92c064f652fb36d925cb975b896 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: 75b584875fc75c437491a5155ef310248e9b9dd4 --- 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,