Former-commit-id: 0edc16769f7e84b74e5fc6a1382e284632567c4c
This commit is contained in:
hiyouga 2024-05-18 16:13:14 +08:00
parent 9fba1bb649
commit 1e867c0fa0

View File

@ -57,9 +57,9 @@ def convert_alpaca(
prompt.append({"role": Role.USER.value, "content": "\n".join(content)}) # "prompt\nquery"
if dataset_attr.kto_tag and isinstance(examples[dataset_attr.kto_tag], bool): # kto example
if dataset_attr.kto_tag and isinstance(examples[dataset_attr.kto_tag][i], bool): # kto example
response = [{"role": Role.ASSISTANT.value, "content": examples[dataset_attr.response][i]}]
if examples[dataset_attr.kto_tag]:
if examples[dataset_attr.kto_tag][i]:
response = response + [{"role": Role.ASSISTANT.value, "content": ""}]
else:
response = [{"role": Role.ASSISTANT.value, "content": ""}] + response