diff --git a/ptuning/main.py b/ptuning/main.py
index 49b08b0..17e18b5 100644
--- a/ptuning/main.py
+++ b/ptuning/main.py
@@ -35,7 +35,6 @@ from transformers import (
     AutoConfig,
     AutoModel,
     AutoTokenizer,
-    AutoTokenizer,
     DataCollatorForSeq2Seq,
     HfArgumentParser,
     Seq2SeqTrainingArguments,