Former-commit-id: 083787dbfe
This commit is contained in:
hiyouga
2023-11-01 23:38:49 +08:00
parent d57f70bf8a
commit 22b3c913e9
5 changed files with 33 additions and 24 deletions

View File

@@ -12,6 +12,8 @@ class DatasetAttr:
dataset_sha1: Optional[str] = None
system_prompt: Optional[str] = None
ranking: Optional[bool] = False
formatting: Optional[Literal["alpaca", "sharegpt"]] = "alpaca"
prompt: Optional[str] = "instruction"
query: Optional[str] = "input"
response: Optional[str] = "output"