Merge branch 'aligner/nemotron5' of https://github.com/NVIDIA/NeMo in…

…to aligner/nemotron5
NVIDIA · Nov 19, 2024 · 6aa111f · 6aa111f
2 parents 3958925 + aee8a89
commit 6aa111f
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/examples/nlp/language_modeling/megatron_mamba_eval.py b/examples/nlp/language_modeling/megatron_mamba_eval.py
@@ -359,9 +359,9 @@ def main(cfg) -> None:
         tdiff = time.perf_counter() - st
         print(f"[Try{i} model.generate took {tdiff} seconds...")
 
-    #print("***************************")
-    #print(response)
-    #print("***************************")
+    # print("***************************")
+    # print(response)
+    # print("***************************")
 
     # Second method of running text generation, call trainer.predict [recommended]
     bs = 2
@@ -375,9 +375,9 @@ def main(cfg) -> None:
         tdiff = time.perf_counter() - st
         print(f"[Try{i} trainer.predict took {tdiff} seconds...")
 
-    #print("***************************")
-    #print(response)
-    #print("***************************")
+    # print("***************************")
+    # print(response)
+    # print("***************************")
 
     # Third method of running text generation, use inference server
     if cfg.server: