This commit is contained in:
hiyouga
2023-11-02 16:51:52 +08:00
parent 083787dbfe
commit dff128c7e3
2 changed files with 5 additions and 2 deletions

View File

@@ -156,5 +156,6 @@ class DataArguments:
dataset_attr.history = dataset_info[name]["columns"].get("history", None)
dataset_attr.ranking = dataset_info[name].get("ranking", False)
dataset_attr.formatting = dataset_info[name].get("formatting", "alpaca")
dataset_attr.system_prompt = prompt_list[i]
self.dataset_list.append(dataset_attr)