From 34db90cc34d592eea1b2663a45b96c295fdfbd2e Mon Sep 17 00:00:00 2001 From: xuanzic Date: Mon, 12 Aug 2024 21:24:28 +0000 Subject: [PATCH] Apply isort and black reformatting Signed-off-by: xuanzic --- .../multimodal/models/multimodal_llm/neva/neva_model.py | 8 ++++++-- 1 file changed, 6 insertions(+), 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 index e36634d23b2c..6218332c2bde 100644 --- 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,