Skip to content

Commit

Permalink
Apply isort and black reformatting
Browse files Browse the repository at this point in the history
Signed-off-by: hemildesai <[email protected]>
  • Loading branch information
hemildesai committed Oct 30, 2024
1 parent 821bb5d commit 14157dc
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions nemo/collections/llm/gpt/data/pre_training.py
Original file line number Diff line number Diff line change
Expand Up @@ -463,6 +463,7 @@ def build_pretraining_datamodule(
torch.distributed.init_process_group(world_size=1, rank=0)

from nemo.utils import logging

logging.info(f"Building {datamodule}")
datamodule.build(
trainer_max_steps=trainer_max_steps,
Expand Down

0 comments on commit 14157dc

Please sign in to comment.