sagemaker.core.jumpstart.configs

sagemaker.core.jumpstart.configs#

This module contains utilites for JumpStart model metadata.

Classes

BaseConfig()

JumpStartConfig(*, model_id[, ...])

Configuration Class for JumpStart.
class sagemaker.core.jumpstart.configs.BaseConfig[source]#

Bases: BaseModel

model_config: ClassVar[ConfigDict] = {'extra': 'forbid', 'frozen': True, 'validate_assignment': True}#

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class sagemaker.core.jumpstart.configs.JumpStartConfig(*, model_id: str, model_version: str | None = None, hub_name: str | None = None, accept_eula: bool | None = False, training_config_name: str | None = None, inference_config_name: str | None = None)[source]#

Bases: BaseConfig

Configuration Class for JumpStart.

model_id#

The model ID of the JumpStart model.

Type:

str

model_version#

The version of the JumpStart model. Defaults to None.

Type:

Optional[str]

hub_name#

The name of the JumpStart hub. Defaults to None.

Type:

Optional[str]

accept_eula#

Whether to accept the EULA. Defaults to None.

Type:

Optional[bool]

training_config_name#

The name of the training configuration. Defaults to None.

Type:

Optional[str]

inference_config_name#

The name of the inference configuration. Defaults to None.

Type:

Optional[str]

accept_eula: bool | None#
hub_name: str | None#
inference_config_name: str | None#
model_config: ClassVar[ConfigDict] = {'extra': 'forbid', 'frozen': True, 'validate_assignment': True}#

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

model_id: str#
model_version: str | None#
training_config_name: str | None#
Contents