このチュートリアルでは、Model Garden を使用して、TPU をベースとする Gemini Enterprise Agent Platform エンドポイントに Gemma 2 2B オープンモデルをデプロイします。オンライン予測用にモデルを配信する前に、モデルをエンドポイントにデプロイする必要があります。モデルのデプロイでは、少ないレイテンシでオンライン予測を提供できるように、モデルに物理リソースを関連付けます。
Gemma 2 2B モデルをデプロイしたら、PredictionServiceClient を使用してトレーニング済みモデルを推論し、オンライン予測を取得します。オンライン予測は、エンドポイントにデプロイされたモデルに対して行われる同期リクエストです。
目標
このチュートリアルでは、次のタスクを行う方法について説明します。
- Model Garden を使用して、Gemma 2 2B オープンモデルを TPU ベースのエンドポイントにデプロイする
PredictionServiceClientを使用してオンライン予測を取得する
費用
このドキュメントでは、課金対象である次の Google Cloudコンポーネントを使用します。
-
A
ct5lp-hightpu-1tmachine type with one TPU_V5 accelerator - Gemini Enterprise Agent Platform prediction and explanation
料金計算ツールを使うと、予想使用量に基づいて費用の見積もりを生成できます。
このドキュメントに記載されているタスクの完了後、作成したリソースを削除すると、それ以上の請求は発生しません。詳細については、クリーンアップをご覧ください。
始める前に
このチュートリアルでは、次の作業が必要です。
- Google Cloud プロジェクトを設定して Agent Platform API を有効にする
- ローカルマシンで以下を行います。
- Google Cloud CLI をインストール、初期化、認証する
- 言語の SDK をインストールする
Google Cloud プロジェクトを設定する
Google Cloud プロジェクトを設定し、Agent Platform API を有効にします。
- Google Cloud アカウントにログインします。 Google Cloudを初めて使用する場合は、 アカウントを作成して、実際のシナリオでの Google プロダクトのパフォーマンスを評価してください。新規のお客様には、ワークロードの実行、テスト、デプロイができる無料クレジット $300 分を差し上げます。
-
In the Google Cloud console, on the project selector page, select or create a Google Cloud project.
Roles required to select or create a project
- Select a project: Selecting a project doesn't require a specific IAM role—you can select any project that you've been granted a role on.
-
Create a project: To create a project, you need the Project Creator role
(
roles/resourcemanager.projectCreator), which contains theresourcemanager.projects.createpermission. Learn how to grant roles.
-
Verify that billing is enabled for your Google Cloud project.
Enable the Agent Platform API, if it is not already enabled.
Roles required to enable APIs
To enable APIs, you need the
serviceusage.services.enablepermission. If you created the project, then you likely already have this permission through the Owner role (roles/owner). Otherwise, you can get this permission through the Service Usage Admin role (roles/serviceusage.serviceUsageAdmin). Learn how to grant roles.-
In the Google Cloud console, on the project selector page, select or create a Google Cloud project.
Roles required to select or create a project
- Select a project: Selecting a project doesn't require a specific IAM role—you can select any project that you've been granted a role on.
-
Create a project: To create a project, you need the Project Creator role
(
roles/resourcemanager.projectCreator), which contains theresourcemanager.projects.createpermission. Learn how to grant roles.
-
Verify that billing is enabled for your Google Cloud project.
Enable the Agent Platform API, if it is not already enabled.
Roles required to enable APIs
To enable APIs, you need the
serviceusage.services.enablepermission. If you created the project, then you likely already have this permission through the Owner role (roles/owner). Otherwise, you can get this permission through the Service Usage Admin role (roles/serviceusage.serviceUsageAdmin). Learn how to grant roles.
Google Cloud CLI を設定する
ローカルマシンで Google Cloud CLI を設定します。
-
Google Cloud CLI をインストールして初期化する
-
すでに gcloud CLI をインストールしている場合は、このコマンドを実行して
gcloudコンポーネントが更新されていることを確認します。gcloud components update
-
gcloud CLI で認証するには、次のコマンドを実行してローカルのアプリケーションのデフォルト認証情報(ADC)ファイルを生成します。コマンドで起動されたウェブフローを使用して、ユーザー認証情報を提供します。
gcloud auth application-default login
詳細については、gcloud CLI 認証構成と ADC 構成をご覧ください。
プログラミング言語の SDK を設定する
このチュートリアルで使用する環境を設定するには、使用する言語の Gemini Enterprise Agent Platform SDK とプロトコル バッファ ライブラリをインストールします。コードサンプルでは、プロトコル バッファ ライブラリの関数を使用して、入力ディクショナリを API が想定する JSON 形式に変換します。
ローカルマシンで、次のいずれかのタブをクリックして、プログラミング言語の SDK をインストールします。
Python
ローカルマシンで、次のいずれかのタブをクリックして、プログラミング言語の SDK をインストールします。
次のコマンドを実行して、Agent Platform SDK for Python をインストールして更新します。
pip3 install --upgrade "google-cloud-aiplatform>=1.64"
次のコマンドを実行して、Python 用のプロトコル バッファ ライブラリをインストールします。
pip3 install --upgrade "protobuf>=5.28"
Node.js
次のコマンドを実行して、Node.js 用の aiplatform SDK をインストールまたは更新します。
npm install @google-cloud/aiplatform
Java
google-cloud-aiplatform を依存関係として追加するには、環境に適したコードを追加します。
BOM ありの Maven
pom.xml に次の HTML を追加します。
<dependencyManagement> <dependencies> <dependency> <artifactId>libraries-bom</artifactId> <groupId>com.google.cloud</groupId> <scope>import</scope> <type>pom</type> <version>26.34.0</version> </dependency> </dependencies> </dependencyManagement> <dependencies> <dependency> <groupId>com.google.cloud</groupId> <artifactId>google-cloud-aiplatform</artifactId> </dependency> <dependency> <groupId>com.google.protobuf</groupId> <artifactId>protobuf-java-util</artifactId> </dependency> <dependency> <groupId>com.google.code.gson</groupId> <artifactId>gson</artifactId> </dependency> </dependencies>
BOM なしの Maven
pom.xml に次の行を追加します。
<dependency> <groupId>com.google.cloud</groupId> <artifactId>google-cloud-aiplatform</artifactId> <version>1.1.0</version> </dependency> <dependency> <groupId>com.google.protobuf</groupId> <artifactId>protobuf-java-util</artifactId> <version>5.28</version> </dependency> <dependency> <groupId>com.google.code.gson</groupId> <artifactId>gson</artifactId> <version>2.11.0</version> </dependency>
BOM なしの Gradle
build.gradle に次の行を追加します。
implementation 'com.google.cloud:google-cloud-aiplatform:1.1.0'Go
次のコマンドを実行して、これらの Go パッケージをインストールします。
go get cloud.google.com/go/aiplatform go get google.golang.org/protobuf go get github.com/googleapis/gax-go/v2
Model Garden を使用して Gemma をデプロイする
Gemma 2 2B モデルを、小規模から中規模のトレーニング用に最適化された ct5lp-hightpu-1t Compute Engine マシンタイプにデプロイします。このマシンには 1 つの TPU v5e アクセラレータが備わっています。TPU を使用してモデルをトレーニングする方法の詳細については、Cloud TPU v5e トレーニングをご覧ください。
このチュートリアルでは、Model Garden のモデルカードを使用して、命令でチューニングされた Gemma 2 2B オープンモデルをデプロイします。具体的なモデル バージョンは gemma2-2b-it です。-it は指示のチューニング済みを表します。
Gemma 2 2B モデルのパラメータ サイズは小さいため、リソース要件が少なく、デプロイの柔軟性が高まります。
Google Cloud コンソールで、[Model Garden] ページに移動します。
[Gemma 2] モデルカードをクリックします。
[デプロイ] をクリックして、[モデルのデプロイ] ペインを開きます。
[モデルのデプロイ] ペインで、次の詳細を指定します。
[デプロイ環境] で [Agent Platform] をクリックします。
[モデルをデプロイする] セクションで、次の操作を行います。
[リソース ID] に
gemma-2b-itを選択します。[モデル名] と [エンドポイント名] のデフォルト値を使用します。例:
- モデル名:
gemma2-2b-it-1234567891234 - エンドポイント名:
gemma2-2b-it-mg-one-click-deploy
エンドポイント名をメモします。コードサンプルで使用されているエンドポイント ID を確認するために必要になります。
- モデル名:
[デプロイの設定] セクションで、次の操作を行います。
[基本] 設定はデフォルト オプションのままにします。
[リージョン] では、デフォルト値のままにするか、リストからリージョンを選択します。リージョンをメモします。コードサンプルに必要になります。
[マシン仕様] で、TPU を使用するインスタンス
ct5lp-hightpu-1t (1 TPU_V5_LITEPOD; ct5lp-hightpu-1t)を選択します。
[デプロイ] をクリックします。デプロイが完了すると、新しいエンドポイントの詳細が記載されたメールが届きます。[オンライン予測] > [エンドポイント] をクリックしてリージョンを選択して、エンドポイントの詳細を確認することもできます。
PredictionServiceClient で Gemma 2 2B を推論する
Gemma 2 2B をデプロイしたら、PredictionServiceClient を使用して「空はなぜ青いの?」というプロンプトのオンライン予測を取得します。
コード パラメータ
PredictionServiceClient コードサンプルでは、以下を更新する必要があります。
PROJECT_ID: プロジェクト ID を確認する手順は次のとおりです。Google Cloud コンソールの [ようこそ] ページに移動します。
ページ上部のプロジェクト選択ツールで、自分のプロジェクトを選択します。
プロジェクト名、プロジェクト番号、プロジェクト ID は [ようこそ] の見出しの後に表示されます。
ENDPOINT_REGION: エンドポイントをデプロイしたリージョンです。ENDPOINT_ID: エンドポイント ID を確認するには、コンソールで確認するか、gcloud ai endpoints listコマンドを実行します。[モデルをデプロイ] ペインからエンドポイント名とリージョンを取得します。コンソール
[オンライン予測] > [エンドポイント] をクリックしてリージョンを選択すると、エンドポイントの詳細を確認できます。
ID列に表示される番号をメモします。gcloud
エンドポイントの詳細を表示するには、
gcloud ai endpoints listコマンドを実行します。gcloud ai endpoints list \ --region=ENDPOINT_REGION \ --filter=display_name=ENDPOINT_NAME出力は次のようになります。
Using endpoint [https://us-central1-aiplatform.googleapis.com/] ENDPOINT_ID: 1234567891234567891 DISPLAY_NAME: gemma2-2b-it-mg-one-click-deploy
サンプルコード
言語のサンプルコードで、PROJECT_ID、ENDPOINT_REGION、ENDPOINT_ID を更新します。次に、コードを実行します。
Python
Vertex AI SDK for Python のインストールまたは更新の方法については、Vertex AI SDK for Python をインストールするをご覧ください。 詳細については、Python API リファレンス ドキュメントをご覧ください。
Node.js
このサンプルを試す前に、クライアント ライブラリを使用した Agent Platform クイックスタートの Node.js の手順に沿って設定を行ってください。
Agent Platform に対する認証を行うには、アプリケーションのデフォルト認証情報を設定します。詳細については、ローカル開発環境の認証の設定をご覧ください。
Java
このサンプルを試す前に、クライアント ライブラリを使用した Agent Platform クイックスタートの Java の手順に沿って設定を行ってください。
Agent Platform に対する認証を行うには、アプリケーションのデフォルト認証情報を設定します。詳細については、ローカル開発環境の認証の設定をご覧ください。
Go
このサンプルを試す前に、クライアント ライブラリを使用した Agent Platform クイックスタートの Go の手順に沿って設定を行ってください。
Agent Platform に対する認証を行うには、アプリケーションのデフォルト認証情報を設定します。詳細については、ローカル開発環境の認証の設定をご覧ください。
クリーンアップ
このチュートリアルで使用したリソースについて、Google Cloud アカウントに課金されないようにするには、リソースを含むプロジェクトを削除するか、プロジェクトを維持して個々のリソースを削除します。
プロジェクトを削除する
- Google Cloud コンソールで [リソースの管理] ページに移動します。
- プロジェクト リストで、削除するプロジェクトを選択し、[削除] をクリックします。
- ダイアログでプロジェクト ID を入力し、[シャットダウン] をクリックしてプロジェクトを削除します。
リソースを個別に削除する
プロジェクトを保持する場合は、このチュートリアルで使用したリソースを削除します。
- モデルのデプロイを解除してエンドポイントを削除する
- Model Registry からモデルを削除する
モデルのデプロイを解除してエンドポイントを削除する
次のいずれかの方法でモデルのデプロイを解除し、エンドポイントを削除します。
コンソール
Google Cloud コンソールで、[オンライン予測]、[エンドポイント] の順にクリックします。
[リージョン] プルダウン リストで、エンドポイントをデプロイしたリージョンを選択します。
エンドポイント名をクリックして詳細ページを開きます。例:
gemma2-2b-it-mg-one-click-deploy。Gemma 2 (Version 1)モデルの行で、 [アクション] をクリックして、次に [エンドポイントからモデルのデプロイを解除] をクリックします。[エンドポイントからモデルのデプロイを解除] ダイアログで [デプロイ解除] をクリックします。
[戻る] ボタンをクリックして [エンドポイント] ページに戻ります。
gemma2-2b-it-mg-one-click-deploy行の末尾にある [アクション] をクリックし、[エンドポイントを削除] を選択します。確認プロンプトで [確認] をクリックします。
gcloud
Google Cloud CLI を使用してモデルのデプロイを解除し、エンドポイントを削除する手順は次のとおりです。
これらのコマンドでは、次のように置き換えます。
- PROJECT_ID は、プロジェクト名に置き換えます。
- LOCATION_ID は、モデルとエンドポイントをデプロイしたリージョンに置き換えます。
- ENDPOINT_ID は、エンドポイント ID に置き換えます。
- DEPLOYED_MODEL_NAME はモデルの表示名に置き換えます。
- DEPLOYED_MODEL_ID はモデル ID に置き換えます。
gcloud ai endpoints listコマンドを実行してエンドポイント ID を取得します。このコマンドを使用すると、プロジェクト内のすべてのエンドポイントのエンドポイント ID が一覧表示されます。このチュートリアルで使用するエンドポイントの ID をメモします。gcloud ai endpoints list \ --project=PROJECT_ID \ --region=LOCATION_ID出力は次のようになります。出力では、ID は
ENDPOINT_IDと呼ばれます。Using endpoint [https://us-central1-aiplatform.googleapis.com/] ENDPOINT_ID: 1234567891234567891 DISPLAY_NAME: gemma2-2b-it-mg-one-click-deploygcloud ai models describeコマンドを実行してモデル ID を取得します。このチュートリアルでデプロイしたモデルの ID をメモします。gcloud ai models describe DEPLOYED_MODEL_NAME \ --project=PROJECT_ID \ --region=LOCATION_ID省略した出力は次のようになります。出力では、ID は
deployedModelIdと呼ばれます。Using endpoint [https://us-central1-aiplatform.googleapis.com/] artifactUri: [URI removed] baseModelSource: modelGardenSource: publicModelName: publishers/google/models/gemma2 ... deployedModels: - deployedModelId: '1234567891234567891' endpoint: projects/12345678912/locations/us-central1/endpoints/12345678912345 displayName: gemma2-2b-it-12345678912345 etag: [ETag removed] modelSourceInfo: sourceType: MODEL_GARDEN name: projects/123456789123/locations/us-central1/models/gemma2-2b-it-12345678912345 ...エンドポイントからモデルのデプロイを解除します。前のコマンドからエンドポイント ID とモデル ID が必要です。
gcloud ai endpoints undeploy-model ENDPOINT_ID \ --project=PROJECT_ID \ --region=LOCATION_ID \ --deployed-model-id=DEPLOYED_MODEL_IDこのコマンドは出力を生成しません。
gcloud ai endpoints deleteコマンドを実行して、エンドポイントを削除します。gcloud ai endpoints delete ENDPOINT_ID \ --project=PROJECT_ID \ --region=LOCATION_IDプロンプトが表示されたら、「
y」と入力して確定します。このコマンドは出力を生成しません。
モデルを削除する
コンソール
Google Cloud コンソールの Agent Platform セクションから [Model Registry] ページに移動します。
[リージョン] プルダウン リストで、モデルをデプロイしたリージョンを選択します。
gemma2-2b-it-1234567891234行の末尾にある [アクション] をクリックします。[モデルを削除] を選択します。
モデルを削除すると、関連するすべてのモデル バージョンと評価が Google Cloud プロジェクトから削除されます。
確認プロンプトで [削除] をクリックします。
gcloud
Google Cloud CLI を使用してモデルを削除するには、モデルの表示名とリージョンを gcloud ai models delete コマンドに指定します。
gcloud ai models delete DEPLOYED_MODEL_NAME \
--project=PROJECT_ID \
--region=LOCATION_ID
DEPLOYED_MODEL_NAME は、モデルの表示名に置き換えます。PROJECT_ID は、プロジェクト名に置き換えます。LOCATION_ID は、モデルをデプロイしたリージョンに置き換えます。
次のステップ
- Gemma オープンモデルについて学習する。
- Gemma の利用規約を確認する。
- オープンモデルについて学習する。
- チューニング済みモデルをデプロイする方法を学習する。
- HuggingFace Textgen Inference(TGI)を使用して Gemma 2 を Google Kubernetes Engine にデプロイする方法を学習する。
PredictionServiceClientの詳細について、Python、Node.js、Java、Go のいずれかの言語で学習する。