This commit is contained in:
hiyouga 2024-05-20 21:43:19 +08:00
parent 9b0f4d7602
commit 7262679666
1 changed files with 2 additions and 1 deletions

View File

@ -34,7 +34,8 @@ class HhRlhfEn(datasets.GeneratorBasedBuilder):
features = datasets.Features( features = datasets.Features(
{ {
"instruction": datasets.Value("string"), "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"))), "history": datasets.Sequence(datasets.Sequence(datasets.Value("string"))),
} }
) )