Revert "Add assert for max_steps to be positive in MegatronGPTSFTModel (

#9110)" This reverts commit 00023c4.
NVIDIA · May 7, 2024 · 6843d86 · 6843d86
1 parent 67d4881
commit 6843d86
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/nemo/collections/nlp/models/language_modeling/megatron_gpt_sft_model.py b/nemo/collections/nlp/models/language_modeling/megatron_gpt_sft_model.py
@@ -74,7 +74,6 @@ def __init__(self, cfg: DictConfig, trainer: Trainer):
             raise ImportError(
                 "Apex was not found. Please see the NeMo README for installation instructions: https://github.com/NVIDIA/NeMo#megatron-gpt."
             )
-        assert trainer.max_steps > 0, "max_steps for SFT can't be negative as its required to build the dataset"
         super().__init__(cfg, trainer=trainer)
         self.sep_id = cfg.get('sep_id', 49704)
         if hasattr(self.cfg.data, "validation_ds"):