[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
NVIDIA · Nov 9, 2022 · 467af91 · 467af91
1 parent 211e68e
commit 467af91
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/nemo/collections/nlp/data/language_modeling/megatron/gpt_prompt_learning_dataset.py b/nemo/collections/nlp/data/language_modeling/megatron/gpt_prompt_learning_dataset.py
@@ -325,7 +325,6 @@ def collate_fn(self, batch, tp_workers=0):
         input_ids = copy.deepcopy(orig_input_ids)
         answer_starts = copy.deepcopy(orig_answer_starts)
 
-
         # Pad taskname_ids to be the same length for the prompt encoder
         if self.virtual_prompt_source == VirtualPromptSource.PROMPT_ENCODER:
             max_taskname_length = max(len(ids) for ids in taskname_ids)

diff --git a/nemo/collections/nlp/data/language_modeling/megatron/t5_prompt_learning_dataset.py b/nemo/collections/nlp/data/language_modeling/megatron/t5_prompt_learning_dataset.py
@@ -200,7 +200,6 @@ def collate_fn(self, batch):
         enc_input = copy.deepcopy(orig_enc_input)
         dec_input = copy.deepcopy(orig_dec_input)
         dec_labels = copy.deepcopy(orig_dec_labels)
-
 
         taskname_ids = self.pad_taskname_ids(taskname_ids)