Add tool_format to overwrite tool formatter template

This commit is contained in:
mMrBun committed 2024-06-22 02:13:23 +08:00
1 parent db9a1912e3
commit 20e2e6fdcb
4 files changed
+10 -3

No files matched your search

+4
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."},