From 81ba7e087c50928bf6398af038ddd9751261ba26 Mon Sep 17 00:00:00 2001
From: zR <2448370773@qq.com>
Date: Sun, 15 Dec 2024 12:43:53 +0800
Subject: [PATCH] add del label

---
 finetune_demo/finetune_vision.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/finetune_demo/finetune_vision.py b/finetune_demo/finetune_vision.py
index e88ef06..e13f317 100644
--- a/finetune_demo/finetune_vision.py
+++ b/finetune_demo/finetune_vision.py
@@ -77,6 +77,7 @@ class Seq2SeqTrainer(_Seq2SeqTrainer):
         with torch.no_grad():
             if self.args.predict_with_generate:
                 output_ids = inputs.pop("output_ids", None)
+                del inputs["labels"]
             loss, generated_tokens, labels = super().prediction_step(
                 model=model,
                 inputs=inputs,