Update train.py

microsoft · Feb 28, 2024 · b4bf755 · b4bf755
1 parent d218529
commit b4bf755
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tuna/src/train.py b/tuna/src/train.py
@@ -23,7 +23,6 @@
 from transformers import Trainer
 
 import utils
-from custom import CustomTrainer
 
 IGNORE_INDEX = -100
 DEFAULT_PAD_TOKEN = "[PAD]"