gemma_tokenizer¶
- torchtune.models.gemma.gemma_tokenizer(path: str, max_seq_len: Optional[int] = None, prompt_template: Optional[Union[str, Dict[Literal['system', 'user','assistant', 'ipython'], Tuple[str, str]]] = None) GemmaTokenizer[源代码]¶
Gemma 的分词器。
- 参数:
path (str) – 分词器的路径
max_seq_len (Optional[int]) – 对单个消息列表进行分词的最大序列长度，超过此长度输入将被截断。默认为 None。
prompt_template (Optional[_TemplateType]) – 可选指定的 prompt 模板。如果是一个字符串，则假定它是
PromptTemplateInterface类的点路径。如果是一个字典，则假定它是一个自定义 prompt 模板，将角色映射到前置/后置标签。
- 返回:
Gemma 分词器的实例化
- 返回类型:
GemmaTokenizer
