프로그래밍 방식으로 사용 가능한 기초 모델에 대한 정보 가져오기

영어 버전 문서로 돌아가기

마지막 업데이트 날짜: 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 라이브러리