[NFC] polish colossalai/booster/plugin/low_level_zero_plugin.py code style (#4256)

Co-authored-by: supercooledith <893754954@qq.com>
1 year ago · abe4f971e0
parent c614a99d28
commit abe4f971e0
1 changed files with 1 additions and 4 deletions
--- a/colossalai/booster/plugin/low_level_zero_plugin.py
+++ b/colossalai/booster/plugin/low_level_zero_plugin.py
@ -208,10 +208,7 @@ class LowLevelZeroPlugin(DPPluginBase):

        if optimizer is not None and \
                not isinstance(optimizer, OptimizerWrapper):
-            optimizer = LowLevelZeroOptimizer(model.unwrap(),
-                                              optimizer,
-                                              self.zero_optim_config,
-                                              self.optim_kwargs,
+            optimizer = LowLevelZeroOptimizer(model.unwrap(), optimizer, self.zero_optim_config, self.optim_kwargs,
                                              self.verbose)

        return model, optimizer, criterion, dataloader, lr_scheduler