minor fix for missing chat attr (#6671)

Signed-off-by: arendu <adithya.r@gmail.com>
NVIDIA · May 18, 2023 · 8aa80ee · 8aa80ee
1 parent 78fe893
commit 8aa80ee
Showing 1 changed file with 1 addition 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
@@ -235,7 +235,7 @@ def _build_dataset(self, data_cfg, is_train=True):
             num_train_samples_per_dataset = [[None]] * len(data_cfg.file_names)
 
         for file_path, num_samples in zip(data_cfg.file_names, num_train_samples_per_dataset):
-            if self.cfg.data.chat:
+            if self.cfg.data.get("chat", False):
                 dataset_cls = GPTSFTChatDataset
             else:
                 dataset_cls = GPTSFTDataset