reformat

axonn-ai · Oct 19, 2024 · 05537a8 · 05537a8
1 parent c08056a
commit 05537a8
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/axonn/intra_layer/fully_connected.py b/axonn/intra_layer/fully_connected.py
@@ -197,7 +197,7 @@ def __init__(
         skip_bias_add=False,
         init_method=None,
         expert_mode=False,
-        **kwargs
+        **kwargs,
     ):
         super(Linear, self).__init__()
 

diff --git a/axonn/intra_layer/overlap_communication.py b/axonn/intra_layer/overlap_communication.py
@@ -128,7 +128,7 @@ def optimize_communication(
     overlap_all_gather=False,
     model_object_for_overlapping_allgathers=None,
     *args,
-    **kwargs
+    **kwargs,
 ):
     global OVERLAP_ALL_REDUCE, OVERLAP_REDUCE_SCATTER
     global ALL_GATHER_ITERATOR

diff --git a/external/models/vit.py b/external/models/vit.py
@@ -135,7 +135,7 @@ def __init__(
         dropout=0.0,
         emb_dropout=0.0,
         inter_layer_parallel_rank=0,
-        G_inter=1
+        G_inter=1,
     ):
         super().__init__()
         image_height, image_width = pair(image_size)