feat(train/training_internlm.py): fix lint error

yingtongxiong · Oct 23, 2023 · b2c1a70 · b2c1a70
1 parent 9cf1ff0
commit b2c1a70
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/internlm/train/training_internlm.py b/internlm/train/training_internlm.py
@@ -50,7 +50,7 @@
 from internlm.solver.optimizer import FSDPadaptOptimizer, HybridZeroOptimizer
 from internlm.solver.optimizer.utils import ParamBcastSyncHandler
 from internlm.train.utils import create_param_groups
-from internlm.utils.common import DummyProfile, get_current_device
+from internlm.utils.common import DummyProfile
 from internlm.utils.logger import get_logger
 from internlm.utils.megatron_timers import megatron_timer as timer
 from internlm.utils.parallel import sync_model_param, sync_model_param_within_tp