Apply isort and black reformatting

Signed-off-by: xuanzic <[email protected]>
xuanzic · Aug 12, 2024 · 34db90c · 34db90c
1 parent f1a59c4
commit 34db90c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/nemo/collections/multimodal/models/multimodal_llm/neva/neva_model.py b/nemo/collections/multimodal/models/multimodal_llm/neva/neva_model.py
@@ -1274,8 +1274,12 @@ def build_train_valid_test_datasets_blend(self):
 
         for each_file_from_path in data_cfg.data_path:
             if is_packed_sequence:
-                train_dataset = NevaPackedSeqDatatset(each_file_from_path, self.cfg.mm_cfg.vision_encoder.get("crop_size"))
-                valid_dataset = NevaPackedSeqDatatset(each_file_from_path, self.cfg.mm_cfg.vision_encoder.get("crop_size"))
+                train_dataset = NevaPackedSeqDatatset(
+                    each_file_from_path, self.cfg.mm_cfg.vision_encoder.get("crop_size")
+                )
+                valid_dataset = NevaPackedSeqDatatset(
+                    each_file_from_path, self.cfg.mm_cfg.vision_encoder.get("crop_size")
+                )
             else:
                 ds_dict = make_supervised_data_module(
                     tokenizer=self.tokenizer,