fix pretrain

Former-commit-id: 75b584875fc75c437491a5155ef310248e9b9dd4
2023-07-13 23:41:54 +08:00
parent 316a02696f
commit 961e6a9ba4
1 changed files with 0 additions and 1 deletions
--- 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,