Merge pull request #4417 from mMrBun/main

Add tool_format parameter to rewrite templates for different function call formats.

Former-commit-id: def6d280db
This commit is contained in:
hoshi-hiyouga
2024-06-24 23:17:55 +08:00
committed by GitHub
6 changed files with 14 additions and 6 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."},