This commit is contained in:
hiyouga
2024-01-19 21:44:32 +08:00
parent b6ec112beb
commit 12043aab9c
4 changed files with 18 additions and 11 deletions

View File

@@ -32,8 +32,8 @@ class Template:
self,
tokenizer: "PreTrainedTokenizer",
messages: List[Dict[str, str]],
system: str,
tools: str,
system: Optional[str] = None,
tools: Optional[str] = None,
cutoff_len: Optional[int] = 1_000_000
) -> Tuple[List[int], List[int]]:
r"""