Skip to content

Commit 6aa111f

Browse files
committed
Merge branch 'aligner/nemotron5' of https://github.com/NVIDIA/NeMo into aligner/nemotron5
2 parents 3958925 + aee8a89 commit 6aa111f

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

examples/nlp/language_modeling/megatron_mamba_eval.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -359,9 +359,9 @@ def main(cfg) -> None:
359359
tdiff = time.perf_counter() - st
360360
print(f"[Try{i} model.generate took {tdiff} seconds...")
361361

362-
#print("***************************")
363-
#print(response)
364-
#print("***************************")
362+
# print("***************************")
363+
# print(response)
364+
# print("***************************")
365365

366366
# Second method of running text generation, call trainer.predict [recommended]
367367
bs = 2
@@ -375,9 +375,9 @@ def main(cfg) -> None:
375375
tdiff = time.perf_counter() - st
376376
print(f"[Try{i} trainer.predict took {tdiff} seconds...")
377377

378-
#print("***************************")
379-
#print(response)
380-
#print("***************************")
378+
# print("***************************")
379+
# print(response)
380+
# print("***************************")
381381

382382
# Third method of running text generation, use inference server
383383
if cfg.server:

0 commit comments

Comments
 (0)