423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-14 23:58:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix #5338

Browse Source 
Former-commit-id: a66ddfea218feefde50fa097d20b4bcbe89ab791
This commit is contained in:
hiyouga 2024-09-03 17:45:17 +08:00
parent 2e1396cd6b
commit 5019c6148b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/data/processors/supervised.py
View File

@ -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],