`nemo_microservices.data_designer.config.models`#

Module Contents#

Classes#

`Distribution`
`DistributionType`
`ImageContext`
`ImageFormat`
`InferenceParameters`
`ManualDistribution`
`ManualDistributionParams`
`Modality`
`ModalityContext`
`ModalityDataType`
`ModelConfig`
`UniformDistribution`
`UniformDistributionParams`

Functions#

load_model_configs

Data#

`DistributionParamsT`
`DistributionT`

API#

class nemo_microservices.data_designer.config.models.Distribution(/, **data: Any)#

Bases: abc.ABC, nemo_microservices.data_designer.config.base.ConfigBase, typing.Generic[nemo_microservices.data_designer.config.models.DistributionParamsT]

distribution_type: nemo_microservices.data_designer.config.models.DistributionType#: None

params: nemo_microservices.data_designer.config.models.DistributionParamsT#: None

abstract sample() → float#

nemo_microservices.data_designer.config.models.DistributionParamsT#: ‘TypeVar(…)’

nemo_microservices.data_designer.config.models.DistributionT: typing_extensions.TypeAlias#: None

class nemo_microservices.data_designer.config.models.DistributionType#

Bases: str, enum.Enum

MANUAL#: ‘manual’

UNIFORM#: ‘uniform’

class nemo_microservices.data_designer.config.models.ImageContext(/, **data: Any)#

Bases: nemo_microservices.data_designer.config.models.ModalityContext

get_context(record: dict) → dict[str, Any]#

image_format: nemo_microservices.data_designer.config.models.ImageFormat | None#: None

modality: nemo_microservices.data_designer.config.models.Modality#: None

class nemo_microservices.data_designer.config.models.ImageFormat#

Bases: str, enum.Enum

GIF#: ‘gif’

JPEG#: ‘jpeg’

JPG#: ‘jpg’

PNG#: ‘png’

WEBP#: ‘webp’

class nemo_microservices.data_designer.config.models.InferenceParameters(/, **data: Any)#

Bases: nemo_microservices.data_designer.config.base.ConfigBase

extra_body: dict[str, Any] | None#: None

property generate_kwargs: dict[str, float | int]#

max_parallel_requests: int#: ‘Field(…)’

max_tokens: int | None#: ‘Field(…)’

temperature: float | nemo_microservices.data_designer.config.models.DistributionT | None#: None

timeout: int | None#: ‘Field(…)’

top_p: float | nemo_microservices.data_designer.config.models.DistributionT | None#: None

class nemo_microservices.data_designer.config.models.ManualDistribution#

Bases: nemo_microservices.data_designer.config.models.Distribution[nemo_microservices.data_designer.config.models.ManualDistributionParams]

distribution_type: nemo_microservices.data_designer.config.models.DistributionType | None#: ‘manual’

params: nemo_microservices.data_designer.config.models.ManualDistributionParams#: None

sample() → float#

class nemo_microservices.data_designer.config.models.ManualDistributionParams(/, **data: Any)#

Bases: nemo_microservices.data_designer.config.base.ConfigBase

values: List[float]#: ‘Field(…)’

weights: List[float] | None#: None

class nemo_microservices.data_designer.config.models.Modality#

Bases: str, enum.Enum

IMAGE#: ‘image’

class nemo_microservices.data_designer.config.models.ModalityContext(/, **data: Any)#

Bases: abc.ABC, pydantic.BaseModel

column_name: str#: None

data_type: nemo_microservices.data_designer.config.models.ModalityDataType#: None

abstract get_context(record: dict) → dict[str, Any]#

modality: nemo_microservices.data_designer.config.models.Modality#: None

class nemo_microservices.data_designer.config.models.ModalityDataType#

Bases: str, enum.Enum

BASE64#: ‘base64’

URL#: ‘url’

class nemo_microservices.data_designer.config.models.ModelConfig(/, **data: Any)#

Bases: nemo_microservices.data_designer.config.base.ConfigBase

alias: str#: None

inference_parameters: nemo_microservices.data_designer.config.models.InferenceParameters#: None

model: str#: None

provider: str | None#: None

class nemo_microservices.data_designer.config.models.UniformDistribution#

Bases: nemo_microservices.data_designer.config.models.Distribution[nemo_microservices.data_designer.config.models.UniformDistributionParams]

distribution_type: nemo_microservices.data_designer.config.models.DistributionType | None#: ‘uniform’

params: nemo_microservices.data_designer.config.models.UniformDistributionParams#: None

sample() → float#

class nemo_microservices.data_designer.config.models.UniformDistributionParams(/, **data: Any)#

Bases: nemo_microservices.data_designer.config.base.ConfigBase

high: float#: None

low: float#: None

nemo_microservices.data_designer.config.models.load_model_configs( model_configs: list[nemo_microservices.data_designer.config.models.ModelConfig] | str | pathlib.Path | None, ) → list[nemo_microservices.data_designer.config.models.ModelConfig]#

nemo_microservices.data_designer.config.models#

Module Contents#

Classes#

Functions#

Data#

API#

`nemo_microservices.data_designer.config.models`#