Add tool_format to overwrite tool formatter template

2024-06-22 02:13:23 +08:00
parent db9a1912e3
commit 20e2e6fdcb
4 changed files with 10 additions and 3 deletions
@@ -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."},