diff --git a/train.py b/train.py
index 9c61acf..b9fe6af 100644
--- a/train.py
+++ b/train.py
@@ -27,7 +27,6 @@ from internlm.train import (
     initialize_optimizer,
     load_new_batch,
     record_current_batch_training_metrics,
-    wrap_FSDP_model,
 )
 from internlm.utils.common import (
     BatchSkipper,