fix autocast

hpcaitech · Dec 5, 2023 · 18ef3b7 · 18ef3b7
1 parent 317f3bd
commit 18ef3b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/colossalai/booster/mixed_precision/fp16_torch.py b/colossalai/booster/mixed_precision/fp16_torch.py
@@ -89,7 +89,7 @@ def __init__(self, module: nn.Module):
         super().__init__(module)
 
     def forward(self, *args, **kwargs):
-        with get_accelerator().autocast()():
+        with get_accelerator().autocast():
             return self.module(*args, **kwargs)