Apply isort and black reformatting

Signed-off-by: hemildesai <[email protected]>
NVIDIA · Jul 18, 2024 · 6306228 · 6306228
1 parent 781c999
commit 6306228
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/nemo/collections/llm/gpt/model/base.py b/nemo/collections/llm/gpt/model/base.py
@@ -28,6 +28,7 @@
 TRAIN_LOSS_REDUCTION = MaskedTokenLossReduction()
 VALIDATION_LOSS_REDUCTION = MaskedTokenLossReduction(validation_step=True)
 
+
 def gpt_data_step(dataloader_iter) -> Dict[str, torch.Tensor]:
     from megatron.core import parallel_state