From d8cff0df986721d1bbb42746d6df717e35a836ad Mon Sep 17 00:00:00 2001
From: jiaweilu <98024616+moonmengmeng@users.noreply.github.com>
Date: Fri, 18 Oct 2024 10:55:34 +0800
Subject: [PATCH] Update trainer.py

The interface to minicpm 2.6 is data, **inputs can't reasonably get data
---
 finetune/trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/finetune/trainer.py b/finetune/trainer.py
index 7da95ed..51f7f76 100644
--- a/finetune/trainer.py
+++ b/finetune/trainer.py
@@ -150,7 +150,7 @@ def prediction_step(
                 else:
                     loss = None
                     with self.compute_loss_context_manager():
-                        outputs = model(**inputs)
+                        outputs = model(data=inputs)
                     if isinstance(outputs, dict):
                         logits = tuple(
                             v for k, v in outputs.items() if k not in ignore_keys