Update low_level_zero_plugin.py

3 months ago · 9e767643dd
parent 3b0df30362
commit 9e767643dd
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"