[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
2024-08-23 05:50:36 +00:00 · 2024-08-23 05:50:36 +00:00 · 3568df498a
parent 9e767643dd
commit 3568df498a
1 changed files with 1 additions and 1 deletions
--- a/colossalai/booster/plugin/low_level_zero_plugin.py
+++ b/colossalai/booster/plugin/low_level_zero_plugin.py
@ -349,7 +349,7 @@ class LowLevelZeroPlugin(DPPluginBase):
        verbose: bool = False,
        cast_inputs: bool = True,
        fp8_communication: bool = False,
-        use_fp8: bool = False
+        use_fp8: bool = False,
    ) -> None:
        super().__init__()
        assert stage in (1, 2), f"LowLevelZeroPlugin only supports stage 1/2 training"