Merge pull request #4417 from mMrBun/main

Add tool_format parameter to rewrite templates for different function call formats.
This commit is contained in:
hoshi-hiyouga authored and GitHub committed 2024-06-24 23:17:55 +08:00
commit def6d280db
6 files changed
+14 -6

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