supervised_data_module cfg fix

Signed-off-by: paul-gibbons <paul@gibbonspaul.com>
NVIDIA · May 6, 2024 · 80736e1 · 80736e1
1 parent c1e5a08
commit 80736e1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nemo/collections/multimodal/data/neva/neva_dataset.py b/nemo/collections/multimodal/data/neva/neva_dataset.py
@@ -1059,8 +1059,8 @@ def make_supervised_data_module(tokenizer, model_cfg) -> Dict:
             conv_template=data_cfg.get("conv_template", "nvgpt"),
             crop_size=crop_size,
             image_token_len=data_cfg.image_token_len,
-            image_folder=data_cfg.image_folder,
-            video_folder=data_cfg.video_folder,
+            image_folder=data_cfg.get('image_folder', None),
+            video_folder=data_cfg.get('video_folder', None),
             image_aspect_ratio=data_cfg.image_aspect_ratio,
             use_im_start_end=getattr(model_cfg.mm_cfg, 'use_im_start_end', False),
             image_processor=image_processor,