可用的基础模型
Amazon SageMaker JumpStart 内置提供先进的公开可用和专有的基础模型,可以自定义并集成到您的生成式人工智能工作流中。
公开可用的基础模型
Amazon SageMaker JumpStart 载入并维护来自第三方来源的开源基础模型。要开始使用其中一种公开可用的模型,请查看JumpStart 基础模型的使用或浏览可用的示例笔记本之一。在公开可用模型的给定示例笔记本中,尝试切换模型 ID,以实验同一模型系列中的不同模型。
有关模型 ID 和使用 SageMaker Python SDK 部署公开可用 JumpStart 基础模型的资源的更多信息,请参阅 通过 SageMaker Python SDK 使用基础模型。
按照定义,基础模型可以根据多种下游任务进行调整。基础模型在大量的通用领域数据上进行训练,可以针对多种使用案例实施相同的模型或进行自定义。选择基础模型时,首先要定义一项特定任务,例如文本生成或映像生成。
公开可用的时间序列预测模型
时间序列预测模型旨在分析和预测一段时间内的序列数据。这些模型可应用于金融、天气预报或能源需求预测等多个领域。Chronos 模型专为时间序列预测任务定制,可根据历史数据规律进行准确预测。
模型名称 | 模型 ID | 模型来源 | 可微调 |
---|---|---|---|
Chronos T5 Small |
autogluon-forecasting-chronos-t5-small |
Amazon |
否 |
Chronos T5 Base |
autogluon-forecasting-chronos-t5-base |
Amazon |
否 |
Chronos T5 Large |
autogluon-forecasting-chronos-t5-large |
Amazon |
否 |
公开可用的文本生成模型
文本生成基础模型可用于多种下游任务,包括文本摘要、文本分类、问题回答、长篇内容生成、简短文案写作、信息提取等。
模型名称 | 模型 ID | 模型来源 | 可微调 |
---|---|---|---|
Alexa TM 20B |
pytorch-textgeneration1-alexa20b |
Amazon |
否 |
Bloom 1b1 |
huggingface-textgeneration-bloom-1b1 |
Hugging Face |
否 |
Bloom 1b7 |
huggingface-textgeneration-bloom-1b7 |
Hugging Face |
否 |
Bloom 3B |
huggingface-textgeneration1-bloom-3b |
Hugging Face |
是 |
Bloom 560m |
huggingface-textgeneration-bloom-560m |
Hugging Face |
否 |
Bloom 7B1 |
huggingface-textgeneration1-bloom-7b1 |
Hugging Face |
是 |
Bloomz 1b1 |
huggingface-textgeneration-bloomz-1b1 |
Hugging Face |
否 |
Bloomz 1b7 |
huggingface-textgeneration-bloomz-1b7 |
Hugging Face |
否 |
BloomZ 3B FP16 |
huggingface-textgeneration1-bloom-3b-fp16 |
Hugging Face |
是 |
Bloomz 560m |
huggingface-textgeneration-bloomz-560m |
Hugging Face |
否 |
BloomZ 7B1 FP16 |
huggingface-textgeneration1-bloomz-7b1-fp16 |
Hugging Face |
是 |
Code Llama 13B |
meta-textgeneration-llama-codellama-13b |
Meta |
是 |
Code Llama 13B Instruct |
meta-textgeneration-llama-codellama-13b-instruct |
Meta |
否 |
Code Llama 13B Python |
meta-textgeneration-llama-codellama-13b-python |
Meta |
是 |
Code Llama 34B |
meta-textgeneration-llama-codellama-34b |
Meta |
是 |
Code Llama 34B Instruct |
meta-textgeneration-llama-codellama-34b-instruct |
Meta |
否 |
Code Llama 34B Python |
meta-textgeneration-llama-codellama-34b-python |
Meta |
是 |
Code Llama 70B |
meta-textgeneration-llama-codellama-70b |
Meta |
是 |
Code Llama 70B Instruct |
meta-textgeneration-llama-codellama-70b-instruct |
Meta |
否 |
Code Llama 70B Python |
meta-textgeneration-llama-codellama-70b-python |
Meta |
是 |
Code Llama 7B |
meta-textgeneration-llama-codellama-7b |
Meta |
是 |
Code Llama 7B Instruct |
meta-textgeneration-llama-codellama-7b-instruct |
Meta |
否 |
Code Llama 7B Python |
meta-textgeneration-llama-codellama-7b-python |
Meta |
是 |
CyberAgentLM2-7B-Chat (CALM2-7B-Chat) |
huggingface-llm-calm2-7b-chat-bf16 |
Hugging Face |
是 |
DistilGPT2 |
huggingface-textgeneration-distilgpt2 |
Hugging Face |
否 |
Dolly V2 12b BF16 |
huggingface-textgeneration-dolly-v2-12b-bf16 |
Hugging Face |
否 |
Dolly V2 3b BF16 |
huggingface-textgeneration-dolly-v2-3b-bf16 |
Hugging Face |
否 |
Dolly V2 7b BF16 |
huggingface-textgeneration-dolly-v2-7b-bf16 |
Hugging Face |
否 |
Dolphin 2.2.1 Mistral 7B |
huggingface-llm-dolphin-2-2-1-mistral-7b |
Hugging Face |
否 |
Dolphin 2.5 Mixtral 8 7B |
huggingface-llm-dolphin-2-5-mixtral-8x7b |
Hugging Face |
否 |
Dolphin 2.7 Mixtral 8 7B |
huggingface-llm-dolphin-2-7-mixtral-8x7b |
Hugging Face |
否 |
EleutherAI GPT Neo 2.7B |
huggingface-llm-eleutherai-gpt-neo-1-3b |
Hugging Face |
否 |
EleutherAI GPT Neo 2.7B |
huggingface-llm-eleutherai-gpt-neo-2-7b |
Hugging Face |
否 |
Falcon 180B BF16 |
huggingface-llm-falcon-180b-bf16 |
Hugging Face |
否 |
Falcon 180B Chat BF16 |
huggingface-llm-falcon-180b-chat-bf16 |
Hugging Face |
否 |
Falcon 40B BF16 |
huggingface-llm-falcon-40b-bf16 |
Hugging Face |
是 |
Falcon 40B Instruct BF16 |
huggingface-llm-falcon-40b-instruct-bf16 |
Hugging Face |
是 |
Falcon 7B BF16 |
huggingface-llm-falcon-7b-bf16 |
Hugging Face |
是 |
Falcon 7B Instruct BF16 |
huggingface-llm-falcon-7b-instruct-bf16 |
Hugging Face |
是 |
Falcon Lite |
huggingface-llm-amazon-falconlite |
Hugging Face |
否 |
Falcon Lite 2 |
huggingface-llm-amazon-falconlite2 |
Hugging Face |
否 |
Falcon RW 1B |
huggingface-llm-tiiuae-falcon-rw-1b |
Hugging Face |
否 |
Flan-T5 Base |
huggingface-text2text-flan-t5-base |
Hugging Face |
是 |
在 Samsum 数据集上微调 Flan-T5 基本模型 |
huggingface-text2text-flan-t5-base-samsum |
Hugging Face |
否 |
Flan-T5 Large |
huggingface-text2text-flan-t5-large |
Hugging Face |
是 |
Flan-T5 Small |
huggingface-text2text-flan-t5-small |
Hugging Face |
是 |
Flan-T5 XL |
huggingface-text2text-flan-t5-xl |
Hugging Face |
是 |
Flan-T5 XXL |
huggingface-text2text-flan-t5-xxl |
Hugging Face |
是 |
Flan-UL2 BF16 |
huggingface-text2text-flan-ul2-bf16 |
Hugging Face |
否 |
Gemma 2B |
huggingface-llm-gemma-2b |
Hugging Face |
是 |
Gemma 2B Instruct |
huggingface-llm-gemma-2b-instruct |
Hugging Face |
是 |
Gemma 7B |
huggingface-llm-gemma-7b |
Hugging Face |
是 |
Gemma 7B Instruct |
huggingface-llm-gemma-7b-instruct |
Hugging Face |
是 |
GPT 2 |
huggingface-textgeneration-gpt2 |
Hugging Face |
否 |
GPT NeoX 20B FP16 |
huggingface-textgeneration2-gpt-neox-20b-fp16 |
Hugging Face |
否 |
GPT NeoXT Chat Base 20B FP16 |
huggingface-textgeneration2-gpt-neoxt-chat-base-20b-fp16 |
Hugging Face |
否 |
GPT-2 XL |
huggingface-textgeneration1-gpt-2-xl |
Hugging Face |
是 |
GPT-J 6B |
huggingface-textgeneration1-gpt-j-6b |
Hugging Face |
是 |
GPT-Neo 1.3B |
huggingface-textgeneration1-gpt-neo-1-3b |
Hugging Face |
是 |
GPT-Neo 125M |
huggingface-textgeneration1-gpt-neo-125m |
Hugging Face |
是 |
GPT-NEO 2.7B |
huggingface-textgeneration1-gpt-neo-2-7b |
Hugging Face |
是 |
Japanese StableLM Instruct Alpha 7B v2 |
model-textgenerationjp-japanese-stablelm-instruct-alpha-7b-v2 |
Hugging Face |
否 |
LightGPT Instruct 6B |
huggingface-textgeneration1-lightgpt |
Hugging Face |
是 |
Lite Llama 460M 1T |
huggingface-llm-ahxt-litellama-460m-1t |
Hugging Face |
否 |
Llama 2 13B |
meta-textgeneration-llama-2-13b |
Meta |
是 |
Llama 2 13B Chat |
meta-textgeneration-llama-2-13b-f |
Meta |
是 |
Llama 2 13B Chat Neuron |
meta-textgenerationneuron-llama-2-13b-f |
Meta |
否 |
Llama 2 13B Neuron |
meta-textgenerationneuron-llama-2-13b |
Meta |
是 |
Llama 2 70B |
meta-textgeneration-llama-2-70b |
Meta |
是 |
Llama 2 70B Chat |
meta-textgeneration-llama-2-70b-f |
Meta |
是 |
Llama 2 70B Chat Neuron |
meta-textgenerationneuron-llama-2-70b-f |
Meta |
否 |
Llama 2 70B Neuron |
meta-textgenerationneuron-llama-2-70b |
Meta |
否 |
Llama 2 7B |
meta-textgeneration-llama-2-7b |
Meta |
是 |
Llama 2 7B Chat |
meta-textgeneration-llama-2-7b-f |
Meta |
是 |
Llama 2 7B Chat Neuron |
meta-textgenerationneuron-llama-2-7b-f |
Meta |
否 |
Llama 2 7B Neuron |
meta-textgenerationneuron-llama-2-7b |
Meta |
是 |
Llama 3 8B |
meta-textgeneration-llama-3-8b |
Meta |
是 |
Llama 3 8B Instruct |
meta-textgeneration-llama-3-8b-instruct |
Meta |
是 |
Llama 3 70B |
meta-textgeneration-llama-3-70b |
Meta |
是 |
Llama 3 70B Instruct |
meta-textgeneration-llama-3-70b-instruct |
Meta |
是 |
Llama Guard 7B |
meta-textgeneration-llama-guard-7b |
Meta |
否 |
Mistral 7B |
huggingface-llm-mistral-7b |
Hugging Face |
是 |
Mistral 7B Instruct |
huggingface-llm-mistral-7b-instruct |
Hugging Face |
否 |
Mistral 7B OpenOrca AWQ |
huggingface-llm-thebloke-mistral-7b-openorca-awq |
Hugging Face |
否 |
Mistral 7B SFT Alpha |
huggingface-llm-huggingfaceh4-mistral-7b-sft-alpha |
Hugging Face |
否 |
Mistral 7B SFT Beta |
huggingface-llm-huggingfaceh4-mistral-7b-sft-beta |
Hugging Face |
否 |
Mistral Lite |
huggingface-llm-amazon-mistrallite |
Hugging Face |
否 |
Mistral Trix V1 |
huggingface-llm-cultrix-mistraltrix-v1 |
Hugging Face |
否 |
Mixtral 8x7B |
huggingface-llm-mixtral-8x7b |
Hugging Face |
是 |
Mixtral 8x7B Instruct |
huggingface-llm-mixtral-8x7b-instruct |
Hugging Face |
是 |
MPT 7B BF16 |
huggingface-textgeneration1-mpt-7b-bf16 |
Hugging Face |
否 |
MPT 7B Instruct BF16 |
huggingface-textgeneration1-mpt-7b-instruct-bf16 |
Hugging Face |
否 |
MPT 7B StoryWriter-65k+ BF16 |
huggingface-textgeneration1-mpt-7b-storywriter-bf16 |
Hugging Face |
否 |
Multilingual GPT |
huggingface-llm-ai-forever-mgpt |
Hugging Face |
否 |
Nous Hermes 2 SOLAR 10.7B |
huggingface-llm-nousresearch-nous-hermes-2-solar-10-7b |
Hugging Face |
否 |
Nous Hermes Llama 2 13B |
huggingface-llm-nousresearch-nous-hermes-llama2-13b |
Hugging Face |
否 |
Nous Hermes Llama 2 7B |
huggingface-llm-nousresearch-nous-hermes-llama-2-7b |
Hugging Face |
否 |
Open Hermes 2 Mistral 7B |
huggingface-llm-teknium-openhermes-2-mistral-7b |
Hugging Face |
否 |
Open LlaMa |
huggingface-textgeneration-open-llama |
Hugging Face |
否 |
Open Llama 7B V2 |
huggingface-llm-openlm-research-open-llama-7b-v2 |
Hugging Face |
否 |
Platypus 2 7B |
huggingface-llm-garage-baind-platypus2-7b |
Hugging Face |
否 |
Pythia 160m Deduped |
huggingface-llm-eleutherai-pythia-160m-deduped |
Hugging Face |
否 |
Pythia 7m Deduped |
huggingface-llm-eleutherai-pythia-70m-deduped |
Hugging Face |
否 |
Quality Controlled Paraphrase Generation |
huggingface-text2text-qcpg-sentences |
Hugging Face |
否 |
RedPajama INCITE Base 3B V1 |
huggingface-textgeneration1-redpajama-incite-base-3B-v1-fp16 |
Hugging Face |
是 |
RedPajama INCITE Base 7B V1 |
huggingface-textgeneration1-redpajama-incite-base-7B-v1-fp16 |
Hugging Face |
是 |
RedPajama INCITE Chat 3B V1 |
huggingface-textgeneration1-redpajama-incite-chat-3B-v1-fp16 |
Hugging Face |
是 |
RedPajama INCITE Chat 7B V1 |
huggingface-textgeneration1-redpajama-incite-chat-7B-v1-fp16 |
Hugging Face |
是 |
RedPajama INCITE Instruct 3B V1 |
huggingface-textgeneration1-redpajama-incite-instruct-3B-v1-fp16 |
Hugging Face |
是 |
RedPajama INCITE Instruct 7B V1 |
huggingface-textgeneration1-redpajama-incite-instruct-7B-v1-fp16 |
Hugging Face |
是 |
Rinna Bilingual GPT NeoX 4B Instruction PPO |
huggingface-llm-bilingual-rinna-4b-instruction-ppo-bf16 |
Hugging Face |
否 |
Rinna Japanese GPT NeoX 3.6B Instruction PPO |
huggingface-llm-rinna-3-6b-instruction-ppo-bf16 |
Hugging Face |
否 |
Star Chat Alpha |
huggingface-llm-huggingfaceh4-starchat-alpha |
Hugging Face |
否 |
Star Chat Beta |
huggingface-llm-huggingfaceh4-starchat-beta |
Hugging Face |
否 |
StarCoder |
huggingface-llm-starcoder |
Hugging Face |
否 |
StarCoderBase |
huggingface-llm-starcoderbase |
Hugging Face |
否 |
T0pp |
huggingface-text2text-bigscience-t0pp |
Hugging Face |
否 |
T5 One Line Summary |
huggingface-text2text-t5-one-line-summary |
Hugging Face |
否 |
Tiny Llama 1.1B |
huggingface-llm-tinyllama-1-1b-intermediate-step-1431k-3 |
Hugging Face |
否 |
Tiny Llama 1.1B Chat V0.6 |
huggingface-llm-tinyllama-tinyllama-1-1b-chat-v0-6 |
Hugging Face |
否 |
Tiny Llama 1.1B Chat V1 |
huggingface-llm-tinyllama-tinyllama-1-1b-chat-v1-0 |
Hugging Face |
否 |
Writer Palmyra Small |
huggingface-llm-writer-palmyra-small |
Hugging Face |
否 |
YARN Mistral 7B 128k |
huggingface-llm-nousresearch-yarn-mistral-7b-128k |
Hugging Face |
否 |
Zephyr 7B Alpha |
huggingface-llm-huggingfaceh4-zephyr-7b-alpha |
Hugging Face |
否 |
Zephyr 7B Beta |
huggingface-llm-huggingfaceh4-zephyr-7b-beta |
Hugging Face |
否 |
要探索最新的文本生成 JumpStart 基础模型,请使用 Amazon SageMaker JumpStart 入门
公开可用的映像生成模型
JumpStart 提供了各种各样的 Stable Diffusion 映像生成基础模型,包括来自 Stability AI 的基础模型,以及 Hugging Face 中用于特定文本到映像任务的预训练模型。如果您需要微调文本到图像基础模型,可以使用 Stability AI 中的 Stable Diffusion 2.1 基础模型。如果您想探索已经接受过特定艺术风格训练的模型,您可以直接在 Amazon SageMaker Studio 用户界面或 SageMaker Studio Classic 用户界面中探索源自 Hugging Face 的众多第三方模型之一。
要探索最新的图像生成 JumpStart 基础模型,请使用 Amazon SageMaker JumpStart 入门
专有基础模型
Amazon SageMaker JumpStart 提供了对第三方提供商的专有基础模型的访问,这些提供商包括 AI21 Labs
要开始使用这些专有模型之一,请参阅JumpStart 基础模型的使用。要使用专有基础模型,您必须先在 AWS Marketplace 中订阅该模型。订阅模型后,在 Studio 或 SageMaker Studio Classic 中找到基础模型。有关更多信息,请参阅 SageMaker JumpStart 预训练模型。
要探索适用于各种使用案例的最新专有基础模型,请参阅 Amazon SageMaker JumpStart 入门