Slim Orca data parsing

Former-commit-id: 7f792dfedebac8ed46eea935fed296bdb50d5f44
This commit is contained in:
Mark Mueller 2024-02-08 17:54:18 +01:00
parent 16d0bf0317
commit 08967e324a

View File

@ -77,7 +77,7 @@ def convert_sharegpt(examples: Dict[str, List[Any]], dataset_attr: "DatasetAttr"
response.append(last_message) response.append(last_message)
outputs["prompt"].append(prompt) outputs["prompt"].append(prompt)
outputs["response"].append(response) outputs["response"].append(response)
if not dataset_attr.system_tag: if n_sys == 0:
outputs["system"].append(examples[dataset_attr.system][i] if dataset_attr.system else "") outputs["system"].append(examples[dataset_attr.system][i] if dataset_attr.system else "")
outputs["tools"].append(examples[dataset_attr.tools][i] if dataset_attr.tools else "") outputs["tools"].append(examples[dataset_attr.tools][i] if dataset_attr.tools else "")