快捷方式

llama2_tokenizer

torchtune.models.llama2.llama2_tokenizer(path: str, max_seq_len: Optional[int] = None, prompt_template: Optional[Union[str, Dict[Literal['system', 'user', 'assistant', 'ipython'], Tuple[str, str]]] = 'torchtune.models.llama2.Llama2ChatTemplate') Llama2Tokenizer[源代码]

Llama2 的分词器。

参数:
  • path (str) – 分词器的路径

  • max_seq_len (Optional[int]) – 对单个消息列表进行分词时的最大序列长度,超过此长度后,输入将被截断。默认为 None。

  • prompt_template (Optional[_TemplateType]) – 可选的指定提示模板。如果为字符串,则假定为 PromptTemplateInterface 类的点路径。如果为字典,则假定为自定义提示模板,将角色映射到前缀/后缀标签。默认为 Llama2ChatTemplate

返回值:

Llama2 分词器的实例化

返回类型:

Llama2Tokenizer

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源