torchtune.models¶
llama3.3¶
来自Llama3家族3.3版本的纯文本模型。
重要提示:在下载之前,您需要在Hugging Face上请求访问权限。
下载 Llama-3.3-70B-Instruct 模型:
tune download meta-llama/Llama-3.3-70B-Instruct --ignore-patterns "original/consolidated.00.pth" --hf-token <HF_TOKEN>
用于创建使用默认70B参数值初始化的Llama3.3模型的构建器。 |
|
用于创建启用LoRA的Llama3.3 70B模型的构建器。 |
|
用于创建启用了QLoRA的Llama3.3 70B模型的构建器。 |
注意
Llama3.3 分词器重用了 llama3_tokenizer 类。
llama3.2¶
来自Llama3家族3.2版本的纯文本模型。
重要提示:在下载之前,您需要在Hugging Face上请求访问权限。
下载 Llama-3.2-1B-Instruct 模型:
tune download meta-llama/Llama-3.2-1B-Instruct --output-dir /tmp/Llama-3.2-1B-Instruct --ignore-patterns "original/consolidated.00.pth" --hf-token <HF_TOKEN>
下载 Llama-3.2-3B-Instruct 模型:
tune download meta-llama/Llama-3.2-3B-Instruct --output-dir /tmp/Llama-3.2-3B-Instruct --ignore-patterns "original/consolidated*" --hf-token <HF_TOKEN>
用于创建使用默认1b参数值初始化的Llama3.2模型的构建器。 |
|
用于创建使用默认3b参数值初始化的Llama3.2模型的构建器。 |
|
用于创建启用LoRA的Llama3.2 1B模型的构建器。 |
|
用于创建启用LoRA的Llama3.2 3B模型的构建器。 |
|
用于创建启用QLoRA的Llama3.2 1B模型的构建器。 |
|
用于创建启用了QLoRA的Llama3.2 3B模型的构建器。 |
注意
Llama3.2 分词器重用了 llama3_tokenizer 类。
llama3.2 视觉¶
来自Llama3家族3.2版本的视觉-语言模型。
重要提示:在下载之前,您需要在Hugging Face上请求访问权限。
下载 Llama-3.2-11B-Instruct 模型:
tune download meta-llama/Llama-3.2-11B-Vision-Instruct --output-dir /tmp/Llama-3.2-11B-Vision-Instruct --hf-token <HF_TOKEN>
Llama 3.2 Vision 11B 模型 |
|
Llama3 Vision 的数据转换(包括分词器)。 |
|
返回一个基于传入配置应用了LoRA的Llama3.2视觉版本( |
|
用于创建启用了QLoRA的Llama3.2 vision 11B模型的构建器。 |
|
构建与Llama3模型相关的解码器,并添加额外的融合交叉注意力层。 |
|
通过将CLIP图像模型与额外的投影头融合模块结合,构建Llama 3.2视觉编码器。 |
|
构建与Llama3模型相关的解码器,并添加额外的融合交叉注意力层。 |
|
通过将CLIP图像模型与额外的投影头融合模块结合,构建Llama 3.2视觉编码器。 |
|
Llama 3.2 Vision 的视觉编码器模型。 |
|
投影变换器用于将预训练冻结编码器(CLIP)的输出适配到预训练的解码器模型。 |
|
此转换结合了Llama 3.2 Vision不同模态的转换。 |
注意
Llama3.2 分词器重用了 llama3_tokenizer 类。
llama3 & llama3.1¶
来自Llama3家族的模型3和3.1。
重要提示:在下载之前,您需要在Hugging Face上请求访问权限。
下载Llama3.1-8B-Instruct模型:
tune download meta-llama/Meta-Llama-3.1-8B-Instruct --output-dir /tmp/Meta-Llama-3.1-8B-Instruct --ignore-patterns "original/consolidated.00.pth" --hf-token <HF_TOKEN>
下载Llama3.1-70B-Instruct模型:
tune download meta-llama/Meta-Llama-3.1-70B-Instruct --output-dir /tmp/Meta-Llama-3.1-70B-Instruct --ignore-patterns "original/consolidated*" --hf-token <HF_TOKEN>
下载Llama3.1-405B-Instruct模型:
tune download meta-llama/Meta-Llama-3.1-405B-Instruct --ignore-patterns "original/consolidated*" --hf-token <HF_TOKEN>
要下载上述模型的Llama3权重,你可以从Meta-Llama-3-8B-Instruct和Meta-Llama-3-70B-Instruct下载,并移除忽略模式标志。
构建与Llama3模型相关的解码器。 |
|
返回一个基于传入配置应用了LoRA的Llama3版本( |
|
用于创建使用默认8b参数值初始化的Llama3模型的构建器。 |
|
用于创建启用LoRA的Llama3 8B模型的构建器。 |
|
用于创建启用QLoRA的Llama3 8B模型的构建器。 |
|
用于创建使用默认70B参数值初始化的Llama3模型的构建器。 |
|
用于创建启用LoRA的Llama3 70B模型的构建器。 |
|
用于创建启用了QLoRA的Llama3 70B模型的构建器。 |
|
Llama3的分词器。 |
|
构建与Llama3.1模型相关的解码器。 |
|
返回一个基于传入配置应用了LoRA的Llama3.1版本( |
|
用于创建使用默认8b参数值初始化的Llama3.1模型的构建器。 |
|
用于创建启用LoRA的Llama3.1 8B模型的构建器。 |
|
用于创建启用QLoRA的Llama3.1 8B模型的构建器。 |
|
用于创建使用默认70B参数值初始化的Llama3.3模型的构建器。 |
|
用于创建启用LoRA的Llama3.3 70B模型的构建器。 |
|
用于创建启用了QLoRA的Llama3.3 70B模型的构建器。 |
|
用于创建使用默认405B参数值初始化的Llama3.1模型的构建器。 |
|
用于创建启用LoRA的Llama3.1 405B模型的构建器。 |
|
用于创建启用了QLoRA的Llama3.1 405B模型的构建器。 |
注意
Llama3.1 分词器重用了 llama3.llama3_tokenizer 构建器类。
llama2¶
所有来自Llama2系列的模型。
重要提示:在下载之前,您需要在Hugging Face上请求访问权限。
下载Llama2-7B模型:
tune download meta-llama/Llama-2-7b-hf --output-dir /tmp/Llama-2-7b-hf --hf-token <HF_TOKEN>
下载Llama2-13B模型:
tune download meta-llama/Llama-2-13b-hf --output-dir /tmp/Llama-2-13b-hf --hf-token <HF_TOKEN>
下载Llama2-70B模型:
tune download meta-llama/Llama-2-70b-hf --output-dir /tmp/Llama-2-70b-hf --hf-token <HF_TOKEN>
构建与Llama2模型相关的解码器。 |
|
返回一个基于传入配置应用了LoRA的Llama2版本( |
|
用于创建Llama2模型的构建器,使用默认的7B参数值进行初始化,参数值来自https://arxiv.org/abs/2307.09288 |
|
用于创建启用LoRA的Llama2 7B模型的构建器。 |
|
用于创建启用了QLoRA的Llama2 7B模型的构建器。 |
|
用于创建Llama2模型的构建器,使用来自https://arxiv.org/abs/2307.09288的默认13B参数值进行初始化。 |
|
用于创建启用LoRA的Llama2 13B模型的构建器。 |
|
用于创建启用了QLoRA的Llama2 13B模型的构建器。 |
|
用于创建Llama2模型的构建器,使用来自https://arxiv.org/abs/2307.09288的默认70B参数值进行初始化 |
|
用于创建启用LoRA的Llama2 70B模型的构建器。 |
|
用于创建启用QLoRA的Llama2 70B模型的构建器。 |
|
Llama2的分词器。 |
|
用于创建Llama2模型的构建器,该模型使用默认的7B参数值进行初始化,参数值来自https://arxiv.org/abs/2307.09288,其中输出层是一个分类层,用于奖励建模,投影到单个类别。 |
|
用于创建启用LoRA的Llama2 7B奖励模型的构建器。 |
|
用于创建启用了QLoRA的Llama2奖励7b模型的构建器。 |
|
提示模板,用于格式化人类和系统提示的聊天数据,并使用在Llama2预训练中使用的适当标签。 |
代码骆驼¶
来自Code Llama家族的模型。
重要提示:在下载之前,您需要在Hugging Face上请求访问权限。
下载CodeLlama-7B模型:
tune download meta-llama/CodeLlama-7b-hf --output-dir /tmp/CodeLlama-7b-hf --hf-token <HF_TOKEN>
用于创建Code-Llama2模型的构建器,使用默认的7B参数值进行初始化,参数值来自https://arxiv.org/pdf/2308.12950.pdf |
|
用于创建启用LoRA的Code-Llama2 7B模型的构建器。 |
|
用于创建启用了QLoRA的Code-Llama2 7B模型的构建器。 |
|
用于创建Code-Llama2模型的构建器,使用默认的13B参数值初始化,参数值来自https://arxiv.org/pdf/2308.12950.pdf |
|
用于创建启用LoRA的Code-Llama2 13B模型的构建器。 |
|
用于创建启用了QLoRA的Code-Llama2 13B模型的构建器。 |
|
用于创建Code-Llama2模型的构建器,使用默认的70B参数值进行初始化,参数值来自https://arxiv.org/pdf/2308.12950.pdf |
|
用于创建启用LoRA的Code-Llama2 70B模型的构建器。 |
|
用于创建启用了QLoRA的Code-Llama2 70B模型的构建器。 |
qwen-2.5¶
来自Qwen2.5家族的0.5B、1.5B、3B、7B、14B、32B、72B大小的模型。
要下载Qwen2.5 1.5B模型,例如:
tune download Qwen/Qwen2.5-1.5B-Instruct --output-dir /tmp/Qwen2_5-1_5B-Instruct
用于创建Qwen2.5模型(基础版或指导版)的构建器,使用默认的0.5B参数值初始化,参数值来自https://huggingface.co/Qwen/Qwen2.5-0.5B-Instruct |
|
用于创建启用了LoRA的Qwen2.5 0.5B模型(基础版或指导版)的构建器。 |
|
用于创建Qwen2.5基础模型的构建器,使用默认的1.5B参数值进行初始化,参数值来自https://huggingface.co/Qwen/Qwen2.5-1.5B |
|
用于创建Qwen2.5指令模型的构建器,使用默认的1.5B参数值初始化,来自https://huggingface.co/Qwen/Qwen2.5-1.5B-Instruct |
|
用于创建启用了LoRA的Qwen2.5 1.5B基础模型的构建器。 |
|
用于创建启用了LoRA的Qwen2.5 1.5B指令模型的构建器。 |
|
用于创建Qwen2.5模型(基础版或指导版)的构建器,使用默认的3B参数值进行初始化,参数值来自https://huggingface.co/Qwen/Qwen2.5-3B-Instruct |
|
用于创建启用LoRA的Qwen2.5 3B模型(基础版或指导版)的构建器。 |
|
用于创建Qwen2.5基础模型的构建器,使用默认的7B参数值初始化,参数值来自https://huggingface.co/Qwen/Qwen2.5-7B |
|
用于创建Qwen2.5指令模型的构建器,使用默认的7B参数值初始化,来自https://huggingface.co/Qwen/Qwen2.5-7B-Instruct |
|
用于创建启用了LoRA的Qwen2.5 7B基础模型的构建器。 |
|
用于创建启用LoRA的Qwen2.5 7B指令模型的构建器。 |
|
用于创建Qwen2.5基础模型的构建器,使用默认的14B参数值初始化,参数值来自https://huggingface.co/Qwen/Qwen2.5-14B |
|
用于创建Qwen2.5指令模型的构建器,使用来自https://huggingface.co/Qwen/Qwen2.5-14B-Instruct的默认14B参数值进行初始化 |
|
用于创建启用了LoRA的Qwen2.5 14B基础模型的构建器。 |
|
用于创建启用LoRA的Qwen2.5 14B指令模型的构建器。 |
|
用于创建Qwen2.5基础模型的构建器,使用默认的32B参数值初始化,参数值来自https://huggingface.co/Qwen/Qwen2.5-32B |
|
用于创建Qwen2.5指令模型的构建器,使用默认的32B参数值初始化,参数值来自https://huggingface.co/Qwen/Qwen2.5-32B-Instruct |
|
用于创建启用LoRA的Qwen2.5 32B基础模型的构建器。 |
|
用于创建启用LoRA的Qwen2.5 32B指令模型的构建器。 |
|
用于创建Qwen2.5基础模型的构建器,使用默认的72B参数值初始化,参数值来自https://huggingface.co/Qwen/Qwen2.5-72B |
|
用于创建Qwen2.5指令模型的构建器,使用默认的72B参数值初始化,参数值来自https://huggingface.co/Qwen/Qwen2.5-72B-Instruct |
|
用于创建启用LoRA的Qwen2.5 72B基础模型的构建器。 |
|
用于创建启用LoRA的Qwen2.5 72B指令模型的构建器。 |
|
Qwen2.5的分词器。 |
qwen-2¶
来自Qwen2系列的0.5B、1.5B和7B大小的模型。
要下载Qwen2 1.5B模型,例如:
tune download Qwen/Qwen2-1.5B-Instruct --output-dir /tmp/Qwen2-1.5B-Instruct
构建与Qwen2模型相关的解码器。 |
|
返回一个基于传入配置应用了LoRA的Qwen2版本( |
|
用于创建Qwen2模型的构建器,该模型使用默认的0.5B参数值初始化,参数值来自https://huggingface.co/Qwen/Qwen2-0.5B-Instruct |
|
用于创建启用LoRA的Qwen2 0.5B模型的构建器。 |
|
用于创建Qwen2模型的构建器,该模型使用默认的1.5B参数值进行初始化,参数值来自https://huggingface.co/Qwen/Qwen2-1.5B-Instruct |
|
用于创建启用LoRA的Qwen2 1.5B模型的构建器。 |
|
用于创建Qwen2模型的构建器,使用默认的7B参数值进行初始化,参数值来自https://huggingface.co/Qwen/Qwen2-7B-Instruct |
|
用于创建启用LoRA的Qwen2 7B模型的构建器。 |
|
Qwen2的分词器。 |
phi-3¶
来自Phi-3 mini系列的模型。
下载Phi-3 Mini 4k指导模型:
tune download microsoft/Phi-3-mini-4k-instruct --output-dir /tmp/Phi-3-mini-4k-instruct --hf-token <HF_TOKEN>
|
|
返回一个应用了LoRA的Phi3版本( |
|
用于创建Phi3 Mini 4K Instruct模型的构建器。 |
|
用于创建启用LoRA的Phi3 Mini(3.8b)模型的构建器。 |
|
用于创建启用QLoRA的Phi3迷你模型的构建器。 |
|
Phi-3 小型分词器。 |
mistral¶
所有来自Mistral AI家族的模型。
重要提示:您需要在Hugging Face上请求访问权限以下载此模型。
下载Mistral 7B v0.1模型:
tune download mistralai/Mistral-7B-v0.1 --output-dir /tmp/Mistral-7B-v0.1 --ignore-patterns "*.safetensors" --hf-token <HF_TOKEN>
构建与mistral模型相关的解码器。 |
|
返回一个基于传入配置应用了LoRA的Mistral版本( |
|
构建一个基础mistral模型,并添加一个分类层。 |
|
返回一个应用了LoRA的Mistral分类器版本(一个 |
|
用于创建Mistral 7B模型的构建器,使用默认的7b参数值进行初始化,参数值来自https://mistral.ai/news/announcing-mistral-7b/ |
|
用于创建启用LoRA的Mistral 7B模型的构建器。 |
|
用于创建启用QLoRA的Mistral模型的构建器。 |
|
用于创建Mistral 7B模型的构建器,使用默认的7b参数值初始化,来源:https://huggingface.co/Ray2333/reward-model-Mistral-7B-instruct-Unified-Feedback,其中输出层是一个分类层,投影到单个类别以进行奖励建模。 |
|
用于创建启用LoRA的Mistral奖励7B模型的构建器。 |
|
用于创建启用了QLoRA的Mistral奖励7B模型的构建器。 |
|
Mistral 模型的标记器。 |
|
根据Mistral的instruct模型进行格式化。 |
gemma¶
来自Gemma系列的2B和7B大小的模型。
重要提示:您需要在Hugging Face上请求访问权限才能使用此模型。
下载Gemma 2B模型(不是Gemma2):
tune download google/gemma-2b --ignore-patterns "gemma-2b.gguf" --hf-token <HF_TOKEN>
下载Gemma 7B模型:
tune download google/gemma-7b --ignore-patterns "gemma-7b.gguf" --hf-token <HF_TOKEN>
构建与gemma模型相关的解码器。 |
|
返回一个基于传入配置应用了LoRA的Gemma版本。 |
|
用于创建Gemma 2B模型的构建器,使用默认的2b参数值初始化,来源:https://blog.google/technology/developers/gemma-open-models/ |
|
用于创建启用LoRA的Gemma 2B模型的构建器。 |
|
用于创建启用QLoRA的Gemma模型的构建器。 |
|
用于创建Gemma 7B模型的构建器,使用默认的7b参数值初始化,来源:https://blog.google/technology/developers/gemma-open-models/ |
|
用于创建启用LoRA的Gemma 7B模型的构建器。 |
|
用于创建启用QLoRA的Gemma模型的构建器。 |
|
Gemma的分词器。 |
gemma2 :¶
来自Gemma家族的2B、9B、27B大小的模型。
重要提示:您需要在Hugging Face上请求访问权限才能使用此模型。
下载Gemma2 2B、9B、27B模型:
tune download google/gemma-2-<MODEL_SIZE>b --ignore-patterns "gemma-2-<MODEL_SIZE>b.gguf" --hf-token <HF_TOKEN>
构建与gemma2模型相关的解码器。 |
|
返回一个基于传入配置应用了LoRA的Gemma版本。 |
|
用于创建Gemma2 2B模型的构建器,使用默认的2b参数值初始化,来源:https://github.com/google/gemma_pytorch/blob/main/gemma/config.py |
|
用于创建启用LoRA的Gemma2 2B模型的构建器。 |
|
用于创建启用QLoRA的Gemma2模型的构建器。 |
|
用于创建Gemma2 9B模型的构建器,使用默认的9b参数值初始化,来源:https://github.com/google/gemma_pytorch/blob/main/gemma/config.py |
|
用于创建启用LoRA的Gemma 9B模型的构建器。 |
|
用于创建启用QLoRA的Gemma模型的构建器。 |
|
用于创建Gemma2 27B模型的构建器,使用默认的27b参数值初始化,来源:https://github.com/google/gemma_pytorch/blob/main/gemma/config.py |
|
用于创建启用LoRA的Gemma2 27B模型的构建器。 |
|
用于创建启用QLoRA的Gemma模型的构建器。 |
|
Gemma的分词器。 |
剪辑¶
使用CLIP编码器支持多模态的视觉组件。
构建与clip模型相关的视觉编码器。 |
|
图像的令牌位置嵌入,对于图像中的每个令牌都是不同的。 |
|
用于平铺图像的标记位置嵌入,每个平铺图像不同,每个标记也不同。 |
|
用于图块的位置嵌入,每个图块不同,图块内的每个标记相同。 |