Apply isort and black reformatting

Signed-off-by: guyueh1 <guyueh1@users.noreply.github.com>
guyueh1 · Nov 30, 2024 · 08ac16c · 08ac16c
1 parent 4ba23e4
commit 08ac16c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/nemo/collections/common/metrics/perf_metrics.py b/nemo/collections/common/metrics/perf_metrics.py
@@ -254,15 +254,15 @@ def _grok(self):
         vocab_size = LLM_VOCAB_SIZE_MAP["grok"]
         return (
             self.layers
-            * self.gbs 
+            * self.gbs
             * self.enc_seq_len
-            * self.hs 
-            * self.hs 
+            * self.hs
+            * self.hs
             * (
                 12
                 + (12 * self.query_groups / self.attention_heads)
                 + (18 * self.moe_router_topk * self.ffn_hs / self.hs)
                 + (12 * self.enc_seq_len / self.hs)
                 + (6 * vocab_size / (self.layers * self.hs))
             )
-        )
+        )