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

for more information, see https://pre-commit.ci
NVIDIA · Nov 20, 2024 · bb1c160 · bb1c160
1 parent f2c4092
commit bb1c160
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/transformer_engine/pytorch/graph.py b/transformer_engine/pytorch/graph.py
@@ -526,7 +526,8 @@ def new_fwd(*user_args, **user_kwargs):
                                     # Only Set the FP8 meta for the modules included by forward
                                     continue
                                 fp8_recipe = FP8GlobalStateManager.get_fp8_recipe()
-                                from transformer_engine.pytorch.attention import DotProductAttention 
+                                from transformer_engine.pytorch.attention import DotProductAttention
+
                                 if (
                                     isinstance(m, DotProductAttention)
                                     and not fp8_recipe.fp8_mha