Add tool_format to overwrite tool formatter template

Former-commit-id: 20e2e6fdcb
This commit is contained in:
mMrBun
2024-06-22 02:00:13 +08:00
parent 98abb5c900
commit c0e005e2ea
4 changed files with 10 additions and 3 deletions

View File

@@ -29,6 +29,10 @@ class DataArguments:
default=None,
metadata={"help": "Which template to use for constructing prompts in training and inference."},
)
tool_format: Optional[str] = field(
default=None,
metadata={"help": "Specifies the tool format template for function calling ."},
)
dataset: Optional[str] = field(
default=None,
metadata={"help": "The name of provided dataset(s) to use. Use commas to separate multiple datasets."},