From f837ae8cb5626fccff10d803d7ffc68080dae878 Mon Sep 17 00:00:00 2001
From: hiyouga <467089858@qq.com>
Date: Tue, 3 Sep 2024 17:45:17 +0800
Subject: [PATCH] fix #5338

Former-commit-id: 69d0acacc38dc90a97ce18447f865f525fff086a
---
 src/llamafactory/data/processors/supervised.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/llamafactory/data/processors/supervised.py b/src/llamafactory/data/processors/supervised.py
index 3981fa12..f488dfca 100644
--- a/src/llamafactory/data/processors/supervised.py
+++ b/src/llamafactory/data/processors/supervised.py
@@ -148,7 +148,7 @@ def preprocess_packed_supervised_dataset(
             logger.warning("Dropped invalid example: {}".format(examples["prompt"][i] + examples["response"][i]))
             continue
 
-        input_ids, labels = _encode_supervised_example(
+        input_ids, labels, _ = _encode_supervised_example(
             prompt=examples["prompt"][i],
             response=examples["response"][i],
             system=examples["system"][i],