fix #3818

2024-05-20 21:43:19 +08:00 · 2024-05-20 21:43:19 +08:00 · 7262679666
parent 9b0f4d7602
commit 7262679666
1 changed files with 2 additions and 1 deletions
--- a/data/hh_rlhf_en/hh_rlhf_en.py
+++ b/data/hh_rlhf_en/hh_rlhf_en.py
@ -34,7 +34,8 @@ class HhRlhfEn(datasets.GeneratorBasedBuilder):
        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"))),
            }
        )