fix hiyouga#3818

zhaop-l · May 20, 2024 · 7262679 · 7262679
1 parent 9b0f4d7
commit 7262679
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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"))),
             }
         )