From 7262679666bf70816911ff2434c7c7ccbca26378 Mon Sep 17 00:00:00 2001
From: hiyouga <467089858@qq.com>
Date: Mon, 20 May 2024 21:43:19 +0800
Subject: [PATCH] fix #3818

---
 data/hh_rlhf_en/hh_rlhf_en.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/data/hh_rlhf_en/hh_rlhf_en.py b/data/hh_rlhf_en/hh_rlhf_en.py
index 1bc18f4f58..aa108fa763 100644
--- a/data/hh_rlhf_en/hh_rlhf_en.py
+++ b/data/hh_rlhf_en/hh_rlhf_en.py
@@ -34,7 +34,8 @@ def _info(self) -> datasets.DatasetInfo:
         features = datasets.Features(
             {
                 "instruction": datasets.Value("string"),
-                "output": datasets.Sequence(datasets.Value("string")),
+                "chosen": datasets.Value("string"),
+                "rejected": datasets.Value("string"),
                 "history": datasets.Sequence(datasets.Sequence(datasets.Value("string"))),
             }
         )