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,