fix to adjust for huggingface#8530 changes

stas00 · Nov 18, 2020 · 6eb8094 · 6eb8094
1 parent b592728
commit 6eb8094
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/seq2seq/finetune.py b/examples/seq2seq/finetune.py
@@ -148,7 +148,7 @@ def _step(self, batch: dict) -> Tuple:
             self.save_readable_batch(batch)
 
         outputs = self(src_ids, attention_mask=src_mask, decoder_input_ids=decoder_input_ids, use_cache=False)
-        lm_logits = outputs[0]
+        lm_logits = outputs["logits"]
         if self.hparams.label_smoothing == 0:
             # Same behavior as modeling_bart.py, besides ignoring pad_token_id
             ce_loss_fct = torch.nn.CrossEntropyLoss(ignore_index=pad_token_id)