From b275e5c562b2f1a19196f9561eddb5c70a128060 Mon Sep 17 00:00:00 2001
From: Ryan_Watson <109671755+HonokaKousaka@users.noreply.github.com>
Date: Tue, 19 Dec 2023 21:27:39 +0800
Subject: [PATCH] Update transformers_summarization_wandb.ipynb

lm_labels are not parameters. labels are the parameters instead.
---
 transformers_summarization_wandb.ipynb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/transformers_summarization_wandb.ipynb b/transformers_summarization_wandb.ipynb
index 2e16982..8188bb4 100644
--- a/transformers_summarization_wandb.ipynb
+++ b/transformers_summarization_wandb.ipynb
@@ -404,7 +404,7 @@
     "        ids = data['source_ids'].to(device, dtype = torch.long)\n",
     "        mask = data['source_mask'].to(device, dtype = torch.long)\n",
     "\n",
-    "        outputs = model(input_ids = ids, attention_mask = mask, decoder_input_ids=y_ids, lm_labels=lm_labels)\n",
+    "        outputs = model(input_ids = ids, attention_mask = mask, decoder_input_ids=y_ids, labels=lm_labels)\n",
     "        loss = outputs[0]\n",
     "        \n",
     "        if _%10 == 0:\n",
@@ -1680,4 +1680,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 1
-}
\ No newline at end of file
+}