fix(train.py): fix ci lint error

2023-09-26 15:51:18 +08:00 · 2023-09-26 15:51:18 +08:00 · c5a7e76ada
parent f3f2511e74
commit c5a7e76ada
1 changed files with 0 additions and 1 deletions
--- 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,