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

NVIDIA#9110)" (NVIDIA#9128) This reverts commit b92e24c.
rohitrango · May 7, 2024 · 47aadb1 · 47aadb1
1 parent 451d459
commit 47aadb1
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"):