プログラムによる使用可能なファウンデーション・モデルに関する情報の取得

資料の英語版に戻る

最終更新: 2024年11月28日

プログラムによる使用可能なファウンデーション・モデルに関する情報の取得

IBM watsonx.aiで使用できるファウンデーション・モデルについて説明します。

watsonx.aiのノートブックから、 watsonx.ai Python ライブラリーで使用可能な関数を使用できます。

watsonx.ai Python ライブラリには、モデルのIDと名前を参照するためのヘルパークラスがあります。詳細は「TextModels参照。

以下のコードでは、TextModelsヘルパークラスを使って、利用可能なモデルのモデルIDをリストアップしています。

Python コード

from ibm_watsonx_ai import APIClient
from ibm_watsonx_ai import Credentials

credentials = Credentials(
                   url = "https://{region}.ml.cloud.ibm.com",
                   api_key = {my-IBM-Cloud-API-key},
                  )

api_client = APIClient(credentials)

api_client.foundation_models.TextModels.show()

出力例

{'GRANITE_13B_CHAT_V2': 'ibm/granite-13b-chat-v2',
'GRANITE_13B_INSTRUCT_V2': 'ibm/granite-13b-instruct-v2',
...
}

例foundation modelの詳細を見る

get_details()を使用すると、簡単な説明やfoundation modelの制限などの詳細を見ることができます。

Python コード

from ibm_watsonx_ai.foundation_models import ModelInference
import json
model_id    = api_client.foundation_models.TextModels.FLAN_T5_XXL
project_id  = {my-project-ID}

model = ModelInference(model_id=model_id, project_id=project_id, api_client=api_client)
model_details = model.get_details()

print( json.dumps( model_details, indent=2 ) )

注:

{region}、 {my-IBM-Cloud-API-key}、および {my-project-ID} を、ご使用の環境で有効な値に置き換えます。

出力例

{
  "model_id": "google/flan-t5-xxl",
  "label": "flan-t5-xxl-11b",
  "provider": "Google",
  "source": "Hugging Face",
  "short_description": "flan-t5-xxl is an 11 billion parameter model based on the Flan-T5 family.",
  ...
}

以下のコードサンプルは、モデルの詳細を表示するために、foundation modelのIDを使用しています。

import json

model_id = api_client.foundation_models.TextModels.FLAN_T5_XXL
model_details = api_client.foundation_models.get_model_specs(model_id)

print( json.dumps( model_details, indent=2 ) )

親トピック: Python ライブラリー