利用可能な基盤モデル

Amazon SageMaker JumpStart では、最新技術を取り入れた公開の基盤モデルや独自の基盤モデルが組み込まれていて、これらをカスタマイズして生成 AI ワークフローに統合することができます。

公開されている基盤モデル

Amazon SageMaker JumpStart は、サードパーティーのソースからオープンソースの基盤モデルをオンボードし、維持しています。公開されているこれらのモデルのいずれかを使い始めるには、「JumpStart 基盤モデルの使用法」を参照するか、利用可能なサンプルノートブックの例のいずれかを調べてください。公開されているモデルのサンプルノートブックで、モデル ID を切り替えて、同じモデルファミリー内のさまざまなモデルを試してみます。

公開されている JumpStart 基盤モデルを SageMaker Python SDK でデプロイする場合のモデル ID とリソースの詳細については、「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	あり
Flan-T5 Base Model (Samsum データセットでファインチューニング済み)	`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 の開始方法」製品詳細ページの [テキスト生成] フィルターを使用します。タスクに基づいた基盤モデルを Amazon SageMaker Studio UI または SageMaker Studio Classic UI で直接調べることもできます。JumpStart では、公開されているテキスト生成モデルの一部のみをファインチューニングに使用できます。詳細については、「Amazon SageMaker Studio Classic で基盤モデルを使用する」を参照してください。

公開されている画像生成モデル

JumpStart では、Stable Diffusion のさまざまな画像生成基盤モデルを提供しています。例えば、Stability AI のベースモデルや、テキストから画像を生成する特定のタスク向けに事前トレーニングされた Hugging Face のモデルがあります。text-to-image の基盤モデルを微調整する必要がある場合は、Stability AI の Stable Diffusion 2.1 ベースを使用できます。特定のアートスタイルで既にトレーニングされているモデルを試す場合は、Amazon SageMaker Studio UI または SageMaker AI Studio Classic UI Hugging Faceでから直接多くのサードパーティーモデルのいずれかを試すことができます。

最新の画像生成 JumpStart 基盤モデルを調べるには、「Amazon SageMaker JumpStart の開始方法」製品詳細ページの [テキストで画像生成] フィルターを使用してください。選択した text-to-image 基盤モデルの使用を開始するには、「JumpStart 基盤モデルの使用法」を参照してください。

独自の基盤モデル

Amazon SageMaker JumpStart は、AI21 Labs、Cohere、LightOn などのサードパーティープロバイダーが提供する独自の基盤モデルへのアクセスを提供します。

これらの独自モデルのいずれかを使い始めるには、「JumpStart 基盤モデルの使用法」を参照してください。独自の基盤モデルを使用するには、まず AWS Marketplaceでモデルをサブスクライブする必要があります。モデルをサブスクライブした後、Studio または SageMaker Studio Classic でその基盤モデルを見つけてください。詳細については、「SageMaker JumpStart の事前トレーニング済みモデル」を参照してください。

さまざまなユースケースに対応する最新の独自の基盤モデルについては、「Amazon SageMaker JumpStart の開始方法」を参照してください。

ブラウザで JavaScript が無効になっているか、使用できません。

AWS ドキュメントを使用するには、JavaScript を有効にする必要があります。手順については、使用するブラウザのヘルプページを参照してください。

ドキュメントの表記規則

基盤モデル

基盤モデルの使用法